歡迎來到Linux教程網
Linux教程網
Linux教程網
Linux教程網
Linux教程網 >> Linux基礎 >> Linux教程 >> Ubuntu 16.04 下 CUDA 8 + cuDNN 5.1安裝

Ubuntu 16.04 下 CUDA 8 + cuDNN 5.1安裝

日期:2017/2/28 13:45:22   编辑:Linux教程

環境:Ubuntu 16.04 64bit

Nvidia GeForce GTX 1070 (驅動Nvidia-Linux-x86_64-367.57)

安裝:CUDA 8.0.44

cuDNN 5.1

顯卡驅動安裝:Ubuntu 16.04 Nvidia 顯卡驅動安裝 見 http://www.linuxidc.com/Linux/2016-12/138907.htm

CUDA是NVIDIA的編程語言平台,想使用GPU就必須要使用cuda。

cuDNN是GPU加速計算深層神經網絡的庫。

下載CUDA

去官網https://developer.nvidia.com/cuda-downloads下載CUDA。

我選擇的是:

Operating System : Linux

Architecture : x86_64

Distribution : Ubuntu

Version : 16.04

Installer Type : runfile(local)

deb裡似乎帶了舊顯卡驅動,安裝完後還要重新裝顯卡驅動,有些麻煩。就用runfile。

安裝cuda

我安裝的時候運行完,顯示安裝cuda_toolkit的時候庫libGLU.so和libXmu.so缺失。或許可以嘗試先安裝:

apt-get install libglu1-mesa libxi-dev libxmu-dev libglu1-mesa-dev

(見Missing recommended library: libGLU.so)

我是先裝Nvidia驅動的,第一次裝cuda的時候總是報Warning : Incomplete installation ...然後提示沒有驅動。我感覺可能是顯卡驅動那邊有問題,於是

sudo update-initramfs -u

重啟。

sudo modprobe nvidia

然後安裝Cuda。

具體參考:this installation did not install the CUDA driver

安裝:

sudo sh ./cuda_8.0.44_linux.run

(可通過sudo sh ./cuda_8.0.44_linux.run --help查看一些參數命令,安裝某一部分或是卸載的參數都可以看到,可能有些情況安裝要帶--override參數)

一大堆條款什麼的,願意看就看吧,不想看直接回車按到底。

輸入accept接受條款。

接下來只有nvidia驅動那裡不要裝,因為我之前已經裝過了顯卡驅動。其他都選是或者默認就可以。

流程:

輸入n。不裝nvidia驅動。

輸入y。安裝cuda toolkit

回車。 使用默認安裝路徑/usr/local/cuda-8.0

輸入y。安裝指向/usr/local/cuda的鏈接

輸入y。安裝Samples。

回車。 使用Samples默認安裝路徑/home/username

環境變量設置

sudo vim ~/.bashrc

在末尾添加

export CUDA_HOME=/usr/local/cuda-8.0

export LD_LIBRARY_PATH=/usr/local/cuda-8.0/lib64:$LD_LIBRARY_PATH

export PATH=/usr/local/cuda-8.0/bin:$PATH

保存退出。

然後刷新。

source ~/.bashrc

動態鏈接庫設置

創建文件:

sudo vim /etc/ld.so.conf.d/cuda.conf

寫入:/usr/local/cuda/lib64

保存之後使其立即生效:

sudo ldconfig -v

檢查

(1)Cuda版本

nvcc --version

我這裡提示nvcc沒安裝。之前說裝toolkit時缺兩個庫,我裝上後重裝了cuda還是這樣,執行cuda.run --silent --toolkit還是沒用。最後只能sudo apt-get install nvidia-cuda-toolkit下一個G了,不知道為什麼。

sudo apt-get install nvidia-cuda-toolkit

nvcc --version

(2)Sample測試

cd /usr/local/cuda/samples

sudo chown -R <username>:<usergroup> .

cd 1_Utilities/deviceQuery

make .

./deviceQuery

輸出一些關於cuda及顯卡的信息則正常。make時可能會因gcc、g++版本過高引發問題,但我沒遇到,解決方法可見底部我的參考。

獲取cuDNN

去官網https://developer.nvidia.com/rdp/cudnn-download下載cuDNN,要注冊NVIDIA賬號。

我選擇的是cuDNN v5.1 -> cuDNN v5.1 Library for Linux

安裝cuDNN

下載cuDNN後解壓

cd cuda/include

sudo cp cudnn.h /usr/local/cuda/include #復制頭文件

cd ../lib64

sudo rm -rf libcudnn.so libcudnn.so.5 #刪除原動態文件

sudo ln -s libcudnn.so.5.1.5 libcudnn.so.5 #生成軟鏈接

sudo ln -s libcudnn.so.5 libcudnn.so #生成軟鏈接

若需要更換cudnn版本,則替換原來的libcudnn*,並重新軟鏈接。

更新鏈接庫:

sudo ldconfig

Ubuntu 14.04 安裝配置CUDA http://www.linuxidc.com/Linux/2014-10/107501.htm

Ubuntu 12.04配置NVIDIA CUDA5.5實錄 http://www.linuxidc.com/Linux/2014-10/107502.htm

Ubuntu安裝Theano+CUDA http://www.linuxidc.com/Linux/2014-10/107503.htm

關於Ubuntu 12.04 下 CUDA5.5 的安裝請參看如下鏈接 Ubuntu 12.04 安裝 CUDA-5.5

Caffe配置簡明教程 ( Ubuntu 14.04 / CUDA 7.5 / cuDNN 5.1 / OpenCV 3.1 ) http://www.linuxidc.com/Linux/2016-09/135016.htm

在Ubuntu 14.04上配置CUDA+Caffe+cuDNN+Anaconda+DIGITS http://www.linuxidc.com/Linux/2016-11/136775.htm

更多Ubuntu相關信息見Ubuntu 專題頁面 http://www.linuxidc.com/topicnews.aspx?tid=2

Copyright © Linux教程網 All Rights Reserved