天天看點

ubuntu16.04 安裝雙版本cuda和cudnn

首先附上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

繼續閱讀