在網上看了那麼多博客,ubuntu安裝cuda時要卸載各種什麼驅動了,什麼舊cuda了,什麼舊包了...,搞到最後電腦系統都出問題了,坑,汗。這裏介紹一種十分簡單地安裝cuda的方法,根本不需要卸載任何東西。另外說明一下如果想從cuda8升級到cuda9也不需要卸載cuda8,直接跟着下面的命令走就行了。
(https://pan.baidu.com/s/107hc3DLVcgh5UXAxqAgfsg 提取碼: 792p)
這裏我自己下載的cuda9和cudnn7,也可以從官網下載,下面有官網鏈接。
一、安裝cuda9
cuda9可以在官網下載(https://developer.nvidia.com/cuda-90-download-archive?target_os=Linux&target_arch=x86_64&target_distro=Ubuntu&target_version=1604&target_type=deblocal)。我選的如下圖中的選項。
我下載cuda9的包名字爲:cuda-repo-ubuntu1604-9-0-local_9.0.176-1_amd64.deb,執行以下命令進行安裝:
sudo dpkg -i cuda-repo-ubuntu1604-9-0-local_9.0.176-1_amd64.deb
sudo apt-get update
sudo apt-cache search cuda
sudo apt-get install cuda-9-0
nvidia-smi
nvcc -V
如果出錯,若找不到nvcc:
The program 'nvcc' is currently not installed. You can install it by typing:sudo apt install nvidia-cuda-toolkit:
那麼可以:
sudo vim ~/.bashrc並在最後一行添加:
export PATH=/usr/local/cuda-9.0/bin:$PATH
export LD_LIBRARY_PATH=/usr/local/cuda-9.0/lib64:$LD_LIBRARY_PATH
保存後再source ~/.bashrc
二、安裝cudnn
先下載 cudnn-9.0-linux-x64-v7.4.1.5.tgz,點擊圖中紅框位置進行下載,需要註冊nvidia賬號(https://developer.nvidia.com/rdp/cudnn-archive)
解壓壓縮包:
tar -xvf cudnn-9.0-linux-x64-v7.4.1.5.tgz
顯示以下信息:
cuda/include/cudnn.h
cuda/NVIDIA_SLA_cuDNN_Support.txt
cuda/lib64/libcudnn.so
cuda/lib64/libcudnn.so.7
cuda/lib64/libcudnn.so.7.4.1
cuda/lib64/libcudnn_static.a
繼續執行以下指令:
sudo cp cuda/include/cudnn.h /usr/local/cuda/include/
sudo cp cuda/lib64/libcudnn* /usr/local/cuda/lib64/
sudo chmod a+r /usr/local/cuda/include/cudnn.h
sudo chmod a+r /usr/local/cuda/lib64/libcudnn*
至此,cuda9和cudnn7安裝完成。
報錯及解決方法:
OSError: libcudnn.so.7.0: cannot open shared object file: No such file or directory錯誤
因爲cuda的路徑可能設置錯了:
sudo ldconfig /usr/local/cuda/lib64
libcudnn.so.7不是軟鏈接
/sbin/ldconfig.real: /usr/local/cuda-9.0/targets/x86_64-linux/lib/libcudnn.so.7 is not a symbolic link
建立軟鏈接
sudo ln -sf /usr/local/cuda-9.0/targets/x86_64-linux/lib/libcudnn.so.7.4.1 /usr/local/cuda-9.0/targets/x86_64-linux/lib/libcudnn.so.7