pytorch安装后，进行验证GPU是否可用

>>> import torch# 查看torch版本>>> torch.__version__'1.2.0'>>> import torchvision# 查看torchvision版本>>> torchvision.__version__'0.4.0'# 查看gpu是否可用>>> torch.cuda.i

陨星落云

3478人浏览 · 2020-07-18 12:14:34

陨星落云 · 2020-07-18 12:14:34 发布

>>> 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'