Linux查看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

使用 PyTorch 查看 CUDA 和 cuDNN 版本

import torch
print(torch.__version__)

print(torch.version.cuda)
print(torch.backends.cudnn.version())