>>> 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'
————————————————
原文链接:https://blog.csdn.net/qq_28368377/article/details/107426997