Ubuntu18.04安装CUDA10、CUDNN

我的系统是Ubuntu18.04、64位，选择CUDA10版本如下：

STEP1：CUDA官网选择适合自己系统的版本下载。

 

 STEP2：安装CUDA

在命令行中键入：

sudo sh cuda_10.2.89_440.33.01_linux.run

如果驱动是独立安装了，一定要选择不安装驱动！选择如下：

 

 安装完成后会提示如下：

这是因为我们在安装CUDA的时候没有选择安装驱动，提示需要安装驱动，忽略就行。

STEP3：添加环境变量：

export CUDA_HOME=/usr/local/cuda 
export PATH=$PATH:$CUDA_HOME/bin 
export LD_LIBRARY_PATH=/usr/local/cuda-10.2/lib64${LD_LIBRARY_PATH:+:${LD_LIBRARY_PATH}}

保存并退出

source ~/.bashrc

STEP4：测试是否安装成功

上面的选项选择安装了CUDA例子，运行其中一个来测试是否安装成功：

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

结果如下就表明成功啦

STEP5：安装CUDNN

在官网下载安装包，需要注册登录才能下载。选择适合自己的版本 

 下载完成后解压并进入文件夹：

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*

在终端查看CUDNN版本：

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

 查看cuda版本

cat /usr/local/cuda/version.txt

 

解决cat /usr/local/cuda/include/cudnn.h | grep CUDNN_MAJOR -A 2无法查看cudnn版本的问题

关于cudnn版本的查看，大部分教程给的操作都是

cat /usr/local/cuda/include/cudnn.h | grep CUDNN_MAJOR -A 2
但是新一点的cudnn都无法再通过这条指令查看版本号了

这是因为，新一些的cudnn版本信息都写在在cudnn_version.h而不是cudnn.h

所以复制h文件的时候，需要在

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 cp cuda/include/cudnn_version.h /usr/local/cuda/include/
查看版本号的指令也改为

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