pytorch - 查看pytorch、cuda、cuDNN版本

可在jupyter中输入如下代码：

import torch
print("Pytorch version：")
print(torch.__version__)
print("CUDA Version: ")
print(torch.version.cuda)
print("cuDNN version is :")
print(torch.backends.cudnn.version())

#输出：
Pytorch version：
1.6.0
CUDA Version: 
10.2
cuDNN version is :
7605