输入cat /usr/local/cuda/include/cudnn_version.h | grep CUDNN_MAJOR -A 2可以查看版本
Cudnn version ≥ 8.+
高版本的 cudnn 包中存的是 lib 不是 lib64 文件夹
1 2 3 4 5 6
# Copy the following files into the CUDA toolkit directory. sudo cp cudnn-*-archive/include/cudnn*.h /usr/local/cuda/include sudo cp -P cudnn-*-archive/lib/libcudnn* /usr/local/cuda/lib64 sudo chmod a+r /usr/local/cuda/include/cudnn*.h /usr/local/cuda/lib64/libcudnn*
查看 cudnn 版本: cat /usr/local/cuda/include/cudnn_version.h | grep CUDNN_MAJOR -A 2
安装测试
查看显卡信息:
1
nvidia-smi
查看Cuda版本:
1
nvcc -V
查看Cudnn版本
cudnn version < 8.0
1
cat /usr/local/cuda/include/cudnn.h | grep CUDNN_MAJOR -A 2
cudnn version ≥ 8.0
1
cat /usr/local/cuda/include/cudnn_version.h | grep CUDNN_MAJOR -A 2