ubuntu16.04安裝多個CUDA、cuDNN版本並自由切換
阿新 • • 發佈:2018-12-10
之前安裝了CUDA8.0,cuDNN5.1,現在同時想用另一個版本,所以安裝兩個版本的CUDA、cuDNN來回切換這樣就不用來回解除安裝安裝了。
1、下載CUDA9.0
進入到下載的檔案所在目錄
cd ~/Downloads
安裝CUDA
sudo sh cuda_9.0.176_384.81_linux.run
注意安裝過程中提示是否安裝驅動的時候選擇n,是否建立軟連線選擇n,其他都選擇是
2、下載cuDNN7
進入到檔案所在目錄
cd ~/Downloads
解壓
tar -xvf cudnn-9.0-linux-x64-v7.tgz
將解壓後的檔案拷貝到相應的目錄下:
sudo cp cuda/include/cudnn.h /usr/local/cuda-9.0/include/ sudo cp cuda/lib64/libcudnn* /usr/local/cuda-9.0/lib64/ sudo chmod a+r /usr/local/cuda-9.0/include/cudnn.h sudo chmod a+r /usr/local/cuda-9.0/lib64/libcudnn*
3、說明
之前安裝CUDA8.0的時候已經新增過環境變數,這裡我們需要對系統變數稍作修改,以後切換CUDA版本的時候只需要修改CUDA目錄軟連線到想使用的CUDA版本目錄就可以了(或者直接修改環境變數,變更CUDA版本的目錄也可以)。接下來分別說明環境變數修改內容和切換步驟:
(1)環境變數修改:
開啟環境變數檔案:
gedit ~/.bashrc
找到其中這幾行:
# added by cuda8.0 export CUDA_HOME=/usr/local/cuda export PATH=$PATH:$CUDA_HOME/bin export LD_LIBRARY_PATH=/usr/local/cuda-8.0/lib64${LD_LIBRARY_PATH:+:${LD_LIBRARY_PATH}}
修改為:
# added by cuda8.0
export CUDA_HOME=/usr/local/cuda
export PATH=$PATH:$CUDA_HOME/bin
export LD_LIBRARY_PATH=/usr/local/cuda/lib64${LD_LIBRARY_PATH:+:${LD_LIBRARY_PATH}}
(2)CUDA版本切換:
刪除原來的軟連線:
sudo rm -rf /usr/local/cuda
建立新的CUDA版本的軟連線:
sudo ln -s /usr/local/cuda-9.0 /usr/local/cuda