pytorch安装成功验证 GPU版验证

import torch
from torch.backends import cudnn

print('------pytorch version-----')
print(torch.version)
print(torch.__version__)

print('--------cuda version----------')
print(torch.cuda.is_available())
print(torch.cuda_version)

print('-----cudnn version -----')
print(cudnn.is_available())
print(cudnn.version())