Ubuntu18.04+GTX1060+显卡驱动+CUDA11.2+cuDNN8.1.1

注意， 如果是在UBUNTU20.04环境下， 安装参考： https://blog.csdn.net/weixin_43229238/article/details/127711920

1. 安装驱动： 使用命令行方式（或者使用界面方式）：

sudo ubuntu-drivers autoinstall

安装好后用命令行查看：

nvidia-smi

2. 安装CUDA11.2.
由于安装CUDA 10.2 总是失败，原因可能时于硬件不匹配或者10.2安装包不完整（虽然下载了好几次）， 最后安装CUDA11.2.
- 使用wget下载；
- cd命令转到到下载目录（略）；
- 执行sh 运行。
运行过程中根据提示选择continue 按钮再输入 accept，安装过程中可以选择不安装driver。
等3-5分钟就装好。我下面贴的图时因为忽略了驱动安装。

sudo wget https://developer.download.nvidia.com/compute/cuda/11.2.0/local_installers/cuda_11.2.0_460.27.04_linux.run
sudo sh cuda_11.2.0_460.27.04_linux.run

添加到环境变量中去：sudo gedit ~/.bashrc，在文件末尾添加：

export PATH="/usr/local/cuda-11.2/bin:$PATH"
export LD_LIBRARY_PATH="/usr/local/cuda-10.2/lib64:$LD_LIBRARY_PATH“”

再使用 source ~/.bashrc 命令激活所修改的这个启动文件。
我用以下命令来显示toolkit 设置的前后变化。

3. 安装cuDNN
下载cuDNN(注意RUN-LIB.DEB包和LIB包是不同的东西)
首先到官网下载cuDNN（https://developer.nvidia.com/rdp/cudnn-archive， 按需注册下载），注意选择版本， 我选的是：图片中显示的8.1.1 RUNTIME

下载后找到文件， 点击运行即可。

再输入以下命令，完成！
这个步骤可能NVIDIA改进过来， 记得之前装过时解压缩下载文件后把所解压得到的lib64和include的文件复制到相应位置。