1. 程式人生 > >ubuntu16.04安裝多個CUDA、cuDNN版本並自由切換

ubuntu16.04安裝多個CUDA、cuDNN版本並自由切換

之前安裝了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