2021-05-12 14:32:11
Ubuntu 16.04 下 CUDA 8 + cuDNN 5.1安裝
環境: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
相關文章