參考博客https://blog.csdn.net/wanzhen4330/article/details/81704474
1. 卸載舊cuda
卸載原來的cuda8.0(注意:不需要卸載顯卡驅動,不要給自己找麻煩):
sudo /usr/local/cuda-8.0/bin/uninstall_cuda_8.0.pl
卸載之后,會發現 /usr/local/cuda-8.0目錄下任然有文件存在,這是cudnn文件,所以還需要將cuda-8.0文件刪除干凈:
sudo rm -rf /usr/local/cuda-8.0
2. 安裝cuda9.0
在nvidia官網下載對應操作系統的CUDA安裝包。
下載cuda9.0的安裝文件 cuda_9.0.176_384.81_linux.run
執行如下命令:
sudo sh cuda_9.0.176_384.81_linux.run
然后用一本書的書角壓住回車鍵,直到服務條款顯示到100%。接着按下面的步驟選擇:
accept
n(不要安裝driver)
y
y
y
安裝完成。
3. cudnn安裝
下載對應版本的cudnn(點擊進去下載),並解壓。這個下載比較煩,需要注冊一個賬戶。
cd $你的cudnn解壓的目錄 sudo cp ./include/cudnn.h /usr/local/cuda/include sudo cp ./lib64/lib* /usr/local/cuda/lib64/ ##復制動態鏈接庫 cd /usr/local/cuda/lib64/ sudo ln -snf libcudnn.so.${VERSION} libcudnn.so ##強制建立軟鏈接
4. 修改環境變量
打開主目錄下的 .bashrc文件,將相關路徑按下面的格式進行修改。
sudo gedit ~/.bashrc
在文本結尾添加
export LD_LIBRARY_PATH=/usr/local/cuda-9.0/lib64:$LD_LIBRARY_PATH export PATH=/usr/local/cuda-9.0/bin:$PATH export CUDA_HOME=/usr/local/cuda-9.0
更新環境變量
source ~/.bashrc
5. 檢查當前的cuda版本
在終端輸入nvcc --version,最后一行顯示Cuda compilation tools, release 9.0, V9.0.130 就說明安裝成功了。