https://docs.nvidia.com/cuda/cuda-toolkit-release-notes/index.html
查看CUDA 版本
cat /usr/local/cuda/version.txt
查看cudnn版本
cat /usr/local/cuda/include/cudnn.h | grep CUDNN_MAJOR -A 2
使用pytorch查看
import torch
print(torch.version.cuda)
print(torch.backends.cudnn.version())