查询pytorch版本并验证GPU是否可用

>>> import torch
# 查看torch版本
>>> torch.__version__
'1.2.0'
>>> import torchvision
# 查看torchvision版本
>>> torchvision.__version__
'0.4.0'
# 查看gpu是否可用
>>> torch.cuda.is_available()
True
# 查看gpu是个数
>>> torch.cuda.device_count()
1
# 查看当前gpu
>>> torch.cuda.current_device()
0
# 查看gpu设备名称
>>> torch.cuda.get_device_name(0)
'GeForce GTX 1650'