CUDA driver version 匹配 CUDA runtime version

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())