使用torch查看cuda和cudnn版本

nvcc -V

import torch

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

我本地的是：

CUDA: 10.0
CUDNN：7605