1 查看CUDA版本号

低版本使用:

nvcc -v

高版本使用:

nvcc --version

查看CUDA版本号。

2 查看cuDNN版本号

旧版本cuDNN使用以下命令查看版本号:

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

而高版本的cuDNN的版本号已经不在cudnn.h中了, 而是在cudnn_version.h中,我们也需要将cudnn_version.h拷贝到 /usr/local/cuda/include中,然后再使用以下命令查看:

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