python查看torch、cuda、cudnn版本

 import torch

1、查看torch版本

print("Hello World, Hello PyTorch {}".format(torch.__version__))
Hello World, Hello PyTorch 1.8.1+cu101

2、查看cuda版本

print("\nCUDA is available:{}, version is {}".format(torch.cuda.is_available(), torch.version.cuda))

CUDA is available:True, version is 10.1

3、查看cudnn

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

4、查看device

print("\ndevice_name: {}".format(torch.cuda.get_device_name(0)))
device_name: TITAN Xp