Ubuntu16.04安装CUDA和cuDNN

1.Ubuntu16.04安装CUDA10.2

具体参考：https://blog.csdn.net/qq_41251963/article/details/109401989 (如果侵权，速删)
环境：Ubuntu 16.04（已安装驱动）
安装： CUDA 10.2 & cuDNN 7.6.5
在终端输入：

wget http://developer.download.nvidia.com/compute/cuda/10.2/Prod/local_installers/cuda_10.2.89_440.33.01_linux.run

1.png
cuda2.png

sudo sh cuda_10.2.89_440.33.01_linux.run

输入accept

cuda10.2-11.png

注意：如果已经安装了driver，这里要去掉，否则会报错！
然后选择 Install

cuda10.2-2.png cuda4.png

查看CUDA版本号：

nvcc -V

如果提示：程序“nvcc”尚未安装。 您可以使用以下命令安装：sudo apt install nvidia-cuda-toolkit

cuda5.png

不要这样安装！
正确的解决方案：

vim ~/.bashrc

在文件最后添加下三行：

export LD_LIBRARY_PATH=/usr/local/cuda-10.2/lib64$LD_LIBRARY_PATH
export PATH=/usr/local/cuda-10.2/bin:$PATH
export CUDA_HOME=usr/local/cuda-10.2$CUDA_HOME

cuda10.2-3.png

不要忘记source一下

source ~/.bashrc

cuda6.png

至此CUDA10.2就安装好了～

2.安装cudnn

cuDNN下载链接：https://developer.nvidia.com/rdp/cudnn-archive
下载 7.6.5版本！！！千万别下载8.0.4版本，巨坑！8.0.4版本不知道啥问题，就安装不上，鬼知道我安装了几遍...乌鱼子...

cudnn1.png
选择第三行的cuDNN Library for Linux下载
cudnn2.png

解压至当前文件夹

tar -xzvf cudnn-10.2-linux-x64-v7.6.5.32.tgz 

运行如下命令：

sudo cp cuda/include/cudnn.h /usr/local/cuda-10.2/include
sudo cp cuda/lib64/libcudnn* /usr/local/cuda-10.2/lib64
sudo chmod a+r /usr/local/cuda-10.2/include/cudnn.h 
sudo chmod a+r /usr/local/cuda-10.2/lib64/libcudnn*

查看版本号看是否安装成功：

cat /usr/local/cuda-10.2/include/cudnn.h | grep CUDNN_MAJOR -A 2

如果没有反应，可以重启一下,或者可以试试输入：

//重启
sudo reboot
//cuDNN8.0之后将版本号放入另一个头文件中
cat /usr/local/cuda-10.2/include/cudnn_version.h | grep CUDNN_MAJOR -A 2

cudnn4.png

至此cuDNN7.6.5就安装好了～

3安装Pytorch1.6

CUDA 10.2与Pytorch1.6匹配

conda install pytorch torchvision cudatoolkit=10.2 -c pytorch