一、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是否可用