一、Linux下查看CUDA和cuDNN版本

https://www.cnblogs.com/jimlau/p/13260269.html

1、查看cuda版本

nvcc -V
cat /usr/local/cuda/version.txt

python脚本

import torch
print(torch.version.cuda)

2、查看cudnn版本

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

如果没有则cudnn可能没有安装
python脚本

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

三、查看torch版本

import torch
print(torch.__version__)
print(torch.cuda.is_available()) # 查看cuda是否可用