首先附上Ubuntu檢視cuda和cudnn版本的指令:
#檢視目前cuda和cudnn版本,</usr/local/cuda>為實際的cuda和cudnn安裝的位置#
-----檢視cuda版本-----
cat /usr/local/cuda/version.txt
-----檢視cudnn版本-----
cat /usr/local/cuda/include/cudnn.h | grep CUDNN_MAJOR -A 2
其次,在NVIDIA官網擷取官方的cuda(.run)和cudnn(.tgz)檔案
-----安裝cuda-----
是否安裝驅動選擇 no
是否建立link選擇 no
其餘參照其他教程即可
安裝完成後在~/.bashrc中添加:
export PATH="$PATH:/usr/local/cuda/bin"
export LD_LIBRARY_PATH="$LD_LIBRARY_PATH:/usr/local/cuda/lib64/"
export LIBRARY_PATH="$LIBRARY_PATH:/usr/local/cuda/lib64"
注意這裡一定要是</usr/local/cuda/>而不是</usr/local/cuda-xxx/>
-----安裝cudnn-----
解壓.tgz得到cuda目錄 #注意所有 x 都代表版本号
cd cuda/include
sudo cp cudnn.h /usr/local/cuda-xxx/include #複制頭檔案到對應cuda版本的檔案夾,如</usr/local/cuda-8.0/include/>
cd cuda/lib64
sudo cp lib* /usr/local/cuda-xxx/lib64/ #複制動态連結庫
cd /usr/local/cuda-xxx/lib64/
sudo rm -rf libcudnn.so libcudnn.so.x #删除原有動态檔案
sudo ln -s libcudnn.so.x.x.x libcudnn.so.x #生成軟銜接
sudo ln -s libcudnn.so.x libcudnn.so
-----切換cuda版本-----
#删除之前建立的軟連結
sudo rm -rf /usr/local/cuda
#建立新 cuda 的軟連結
sudo ln -s /usr/local/cuda-xxx /usr/local/cuda