Unbuntu18.04 安装CUDA10.1 + CUDNN7.6

前置条件：已经安装好显卡驱动(2080TI)，具体参考另外一篇博客。

安装环境：Ubuntu18.04 + CUDA10.1 + CUDNN7.6

一.安装CUDA10.1

1.版本选择

查看驱动满足的CUDA版本https://docs.nvidia.com/cuda/cuda-toolkit-release-notes/

这里我们选择安装CUDA10.1

2.下载CUDA10.1

wget http://developer.download.nvidia.com/compute/cuda/10.1/Prod/local_installers/cuda_10.1.243_418.87.00_linux.run

3.安装CUDA

sudo sh cuda_10.1.243_418.87.00_linux.run

1）选择continue

2）输入accept

3）已经安装过显卡驱动，不需要再次安装，使用空格键取消勾选，然后选择install。

4）出现以下结果，则说明安装成功。

 4.添加环境变量

1）配置CUDA的环境变量

vim ~/.bashrc ，在末尾添加如下内容

export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:/usr/local/cuda-10.1/lib64

export PATH=$PATH:/usr/local/cuda-10.1/bin

export CUDA_HOME=$CUDA_HOME:/usr/local/cuda-10.1

2）使配置文件生效

source ~/.bashrc

5. 测试CUDA安装成功的Samples例子

依次执行如下命令

cd /usr/local/cuda-10.1/samples/1_Utilities/deviceQuery

sudo make

./deviceQuery

Result = PASS 则说明安装成功

6. 查看CUDA版本

nvcc -V

二.安装CUDNN7.6

1.版本选择

选择与CUDA对应的CUDNN版本，官方下载地址：https://developer.nvidia.com/rdp/cudnn-archive ，需要注册登录。如下图所示：

2. 解压

tar –zxvf cudnn-10.1-linux-x64-v7.6.5.32.tgz

3. 执行如下命令

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
sudo chmod a+r /usr/local/cuda/lib64/libcudnn*
#重启系统
sudo reboot

4. 验证是否安装成功

cat /usr/local/cuda/include/cudnn.h | grep CUDNN_MAJOR -A 2

出现CUDNN的版本信息，则说明安装成功。