1. 程式人生 > >ubuntu16.04 安裝雙版本cuda和cudnn

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