環境: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