之前安裝了CUDA8.0,cuDNN5.1温亲,現(xiàn)在同時(shí)想用另一個(gè)版本棚壁,所以安裝兩個(gè)版本的CUDA、cuDNN來(lái)回切換這樣就不用來(lái)回卸載安裝了栈虚。
1袖外、下載CUDA9.0
進(jìn)入到下載的文件所在目錄
cd ~/Downloads
安裝CUDA
sudo sh cuda_9.0.176_384.81_linux.run
注意安裝過(guò)程中提示是否安裝驅(qū)動(dòng)的時(shí)候選擇n,是否創(chuàng)建軟連接選擇n魂务,其他都選擇是
2曼验、下載cuDNN7
進(jìn)入到文件所在目錄
cd ~/Downloads
解壓
tar -xvf cudnn-9.0-linux-x64-v7.tgz
將解壓后的文件拷貝到相應(yīng)的目錄下:
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、說(shuō)明
之前安裝CUDA8.0的時(shí)候已經(jīng)添加過(guò)環(huán)境變量粘姜,這里我們需要對(duì)系統(tǒng)變量稍作修改鬓照,以后切換CUDA版本的時(shí)候只需要修改CUDA目錄軟連接到想使用的CUDA版本目錄就可以了(或者直接修改環(huán)境變量,變更CUDA版本的目錄也可以)孤紧。接下來(lái)分別說(shuō)明環(huán)境變量修改內(nèi)容和切換步驟:
(1)環(huán)境變量修改:
打開(kāi)環(huán)境變量文件:
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版本切換:
刪除原來(lái)的軟連接:
sudo rm -rf /usr/local/cuda
建立新的CUDA版本的軟連接:
sudo ln -s /usr/local/cuda-9.0 /usr/local/cuda
4豺裆、查看cuda、cudnn版本
#cuda版本
cat /usr/local/cuda/version.txt
#cudnn版本
cat /usr/local/cuda/include/cudnn.h | grep CUDNN_MAJOR -A 2
————————————————
版權(quán)聲明:本文為CSDN博主「生命長(zhǎng)跑」的原創(chuàng)文章
原文鏈接:https://blog.csdn.net/qq_30683995/article/details/82745246