【Linux】查看CUDA和cuDNN版本

一、查看CUDA版本

方法一：

nvcc -version 
或者
nvcc -V

方法二：

cat /usr/local/cuda/version.txt

二、查看cuDNN版本

使用 PyTorch 查看 CUDA 和 cuDNN 版本

import torch
print(torch.__version__)

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

参考：
https://www.cnblogs.com/jimlau/p/13260269.html