pytorch环境下查看gpu是否可用

(36条消息) yolov5设置GPU – CSDN

python
#输入库
import torch
#查看版本
print(torch.__version__)
#查看gpu是否可用
torch.cuda.is_available()
#返回设备gpu个数
torch.cuda.device_count()
#退出python
quit()