Ubuntu16.04安装CUDA10.2和cuDNN过程详细记录

 进入官网，找到对应的10.2版本，选择要安装的平台；

 在终端执行命令：

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

 

继续执行命令：

sudo sh cuda_10.2.89_440.33.01_linux.run

 输入accept

注意：如果你已经安装了driver，这里需要去掉，否则会报错！！！

点击安装：

稍等一会：

输入nvcc -V查看

发现还是10.0版本，因此需要修改一下环境变量：

 sudo gedit ~/.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

 加入新变量：

 

 

保存之后，为了使这个修改立即生效，还需要输入如下代码（如果不执行 source 命令，则需重启系统才能生效）：

source ~/.bashrc

这样就变成10.2啦！可以安装多个版本的cuda，使用时切换环境变量即可！！！

 

cuDNN下载安装：https://developer.nvidia.com/rdp/cudnn-download

需要登录下载：

 

 选则第一个：

 

通过命令解压文件：

tar -xzvf cudnn-10.2-linux-x64-v8.0.4.30.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*

 

使用查看安装版本：

dpkg -l | grep cudnn

发现问题，版本不对，还是之前版本。

 

 

在cudNN官网上下载：

使用命令安装：

sudo dpkg -i

 

 

 

 

再次查看版本：

 

 

 

 

安装Pytorch：

 

 

测试：