在网上看了那么多博客,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