查看CUDA和CuDNN版本

查看CUDA版本：

nvcc --version 或 nvcc -V

也可以通过安装目录查看：

cat /usr/local/cuda/version.txt

查看CuDNN版本：

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

CuDNN安装

安装了cuda之后，需要安装cudnn加速。

我们可以去官网下载，进入官网后找到相应的资源，选择选择 cuDNN Library for Linux下载。

下载完成后我们进行解压：

tar -xzvf cudnn-10.1-linux-x64-v7.6.5.32.tgz

当面目录会产生cuda文件夹，不进入文件夹，继续输入如下命令：

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*

这样就完成了！

我们可以使用上面提到了命令进行检查安装是否成功：

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