Ubuntu 20.04 安装CUDA Toolkit和cuDNN

Ubuntu 20.04 安装 cuda-toolkit-11.0 和 cudnn-8.0.5

1、首先下载安装cuda驱动

按win + A键，打开所有程序，然后找到Software & Update，点击之后选择Additional Drivers，选择其中一个，然后点击Apply changes，如下图所示：

然后重启系统，在终端中输入nvidia-smi，查看驱动是否安装成功，出来这种表格就算成功了。

2、下载并安装 CUDA Toolkit

这里下载的是CUDA Toolkit 11.0.3版本。

wget https://developer.download.nvidia.com/compute/cuda/11.0.3/local_installers/cuda_11.0.3_450.51.06_linux.run
sudo sh cuda_11.0.3_450.51.06_linux.run 

注意安装的时候，会提示已经有驱动了，这时我们不选择Abort，选择继续安装，然后填入accept。
紧接着，当选择安装哪些组件的时候，需要取消勾选cuda driver，不然会安装失败。安装成功结果如下：

然后添加环境变量到 ~/.bashrc 文件的末尾：

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

保存退出后，source ~/.bashrc是环境变量生效。

测试一下装的cuda有没有生效：

cd /usr/local/cuda/samples/1_Utilities/deviceQuery
sudo make
./deviceQuery

测试成功的话输出如下信息：

3、下载cuDNN

从官网下载cudnn-11.0-linux-x64-v8.0.5.39.tgz，并进行如下操作：

tar zxvf cudnn-11.0-linux-x64-v8.0.5.39.tgz 
sudo cp cuda/include/cudnn* /usr/local/cuda/include
sudo cp cuda/lib64/libcudnn* /usr/local/cuda/lib64
sudo chmod a+r /usr/local/cuda/include/cudnn* /usr/local/cuda/lib64/libcudnn*

Enjoy it！！

参考链接

Ubuntu 20.04 安装 CUDA Toolkit 的三种方式