1.安装显卡驱动
简单一步在ubuntu18.04下nvidia driver的安装
直接通过Ubuntu自带UI界面安装，应用更改自动安装，重启生效
卸载CUDA和cuDNN看这
2.安装CUDA10.1
官网下载地址
运行.run文件
sudo sh cuda_10.1.105_418.39_linux.run
因为已经安装了NVIDIA 显卡驱动，所以这里不勾选安装驱动
等待片刻，出现安装成功
添加环境变量
sudo vim ~/.bashrc
export CUDA_HOME=/usr/local/cuda-10.1
export PATH=/usr/local/cuda-10.1/bin${PATH:+:$PATH}
export LD_LIBRARY_PATH=/usr/local/cuda-10.1/lib64${LD_LIBRARY_PATH:+:${LD_LIBRARY_PATH}}
验证
nvcc --version
3.安装cudnn7.6.2
官网下载地址
下载cuDNN Library for Linux，解压，在解压的文件夹中打开终端
复制文件到目录
sudo cp include/cudnn.h /usr/local/cuda/include/
sudo cp lib64/lib* /usr/local/cuda/lib64/
建立软链接
在/usr/local/cuda/lib64目录下打开终端，执行如下指令：
sudo chmod +r libcudnn.so.7.6.2
sudo ln -sf libcudnn.so.7.6.2 libcudnn.so.7
sudo ln -sf libcudnn.so.7 libcudnn.so
sudo ldconfig
查看当前cudnn的版本
cat /usr/local/cuda/include/cudnn.h | grep CUDNN_MAJOR -A 2
