下载显卡驱动
CUDA版本需与驱动版本对应：https://docs.nvidia.com/cuda/cuda-toolkit-release-notes/index.html
Ubuntu推荐版本
可以在这里下载
#也可以执行以下命令,安装推荐的nvidia版本
sudo ubuntu-drivers autoinstall
#查看
nvidia-smi
CUDA
CUDA安装
CUDA下载地址： https://developer.nvidia.com/cuda-toolkit-archive
以cuda10.1为例，在官网下载cuda_10.1.243_418.87.00_linux.run文件
提示接受条款，输入accept接受；
- 输入n不安装；
- 接着输入y同意安装cuda工具；
- 输入y安装指向/usr/local/cuda的符号链接；（ 可能会有问题 ）
- 输入y安装samples，可以测试是否安装成功；
- 回车确认samples安装的默认路径，/home/用户名，安装路径测试完可以删除。
添加环境变量
sudo apt-get install gedit
sudo gedit /etc/profile
在末尾添加：
export PATH=/usr/local/cuda-10.0/bin:$PATH
export LD_LIBRARY_PATH=/usr/local/cuda-10.0/lib64:/usr/local/cuda-10.0/extras/CUPTI/lib64:$LD_LIBRARY_PATH
保存文件后重启系统
CUDA测试
nvcc --version，输出信息说明安装正确
编译samples
默认路径为~/NVIDIA_CUDA-10.0Samples，命令行cd进入后make，这里需要等一段时间。
之后会生成可执行文件在~/NVIDIA_CUDA-10.0_Samples/bin/x84_64/linux/release，进入目录后输入./deviceQuery
CUDA卸载
cuda10.0及以下的卸载：
cd /usr/local/cuda-xx.x/bin/
sudo ./uninstall_cuda_xx.x.pl
sudo rm -rf /usr/local/cuda-xx.x
cuda10.0及以上的卸载：
cd /usr/local/cuda-xx.x/bin/
sudo ./cuda-uninstaller
sudo rm -rf /usr/local/cuda-xx.x
最后边加了一句sudo rm -rf /usr/local/cuda-xx.x，这是因为一般情况下cuda都配置了cudnn，在运行卸载程序时只会卸载cuda而不会一并删除cudnn的文件。因为cudnn文件还在的缘故，自己的cuda-xx.x文件夹仍然在，需要手动删除。
所以如果自己要卸载的cuda没有配置cudnn，那么cuda-xx.x文件夹在卸载完成后会被自动删除。
cuDNN
cuDNN安装
官网下载： https://developer.nvidia.com/rdp/cudnn-archive ，下载时注意选择与CUDA适配的版本，也是.run文件
tar -xvzf cudnn-10.1-linux-x64-vxxxx.tar.gz
sudo cp cuda/include/cudnn.h /usr/local/cuda/include
sudo cp cuda/lib64/libcudnn* /usr/local/cuda/lib64
sudo chmod a+r /usr/local/cuda/include/cudnn.h /usr/local/cuda/lib64/libcudnn*
cuDNN卸载
sudo rm -rf /usr/local/cuda/include/cudnn.h
sudo rm -rf /usr/local/cuda/lib64/libcudnn*
不是符号链接的问题
在安装好cuDNN之后，运行 sudo ldconfig ，是否会出现
/sbin/ldconfig.real: /usr/local/cuda-10.1/targets/x86_64-linux/lib/libcudnn_ops_infer.so.8 is not a symbolic link
如果有输入 sudo ldconfig -v
在Terminal中找到
libcudnn_ops_infer.so.8 -> libcudnn_ops_infer.so.8.0.5
运行
sudo ln -sf /usr/local/cuda-10.1/targets/x86_64-linux/lib/libcudnn_ops_infer.so.8.0.5 /usr/local/cuda-10.1/targets/x86_64-linux/lib/libcudnn_ops_infer.so.8
然后重启系统
参考链接：
https://blog.csdn.net/frothmoon/article/details/107984882?spm=1001.2014.3001.5501
https://blog.csdn.net/langb2014/article/details/54376716