cuda的下载地址:https://developer.nvidia.com/cuda-toolkit
cudnn的下载地址:https://developer.nvidia.com/rdp/cudnn-download
下载相应的版本就可以啦,我这边下载的cuda是deb文件。
1.cuda的安装与卸载
安装(以cuda9.0为例):
sudo dpkg -i cuda-repo-ubuntu1604-9-0-local-rc_9.0.103-1_amd64.deb
sudo apt-key add /var/cuda-repo-9-0-local-rc/7fa2af80.pub
sudo apt-get update
sudo apt-get install cuda-9.0 #(这里需要指定是9.0的版本)
值得注意的是:如果直接使用 “sudo apt-get install cuda”来装cuda,安装的cuda的最新版本,而不是cuda9.0.
查看cuda的版本:
cat /usr/local/cuda/version.txt
卸载:(网上有人说cuda中提供了自卸载的文件,但是笔者怎么也找不到,下面提供一种卸载方式)
sudo apt-get remove cuda
sudo apt autoremove
ls
sudo rm -rf cuda-9.0/
cd /var/
sudo rm -rf cuda-repo-9-0-local-rc/
2.cudnn的安装与卸载
安装:
cudnn下载之后是一个压缩文件,解压后得到一个名为cuda的文件夹
sudo cp cuda/include/cudnn.h /usr/local/cuda/include/
sudo cp cuda/lib64/libcudnn* /usr/local/cuda/lib64/ -d
sudo chmod a+r /usr/local/cuda/include/cudnn.h
sudo chmod a+r /usr/local/cuda/lib64/libcudnn*
卸载:
sudo rm -rf /usr/local/cuda/include/cudnn.h
sudo rm -rf /usr/local/cuda/lib64/libcudnn*
查看cudnn版本:
cat /usr/local/cuda/include/cudnn.h | grep CUDNN_MAJOR -A 2