pytorch环境下查看gpu是否可用

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

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

标签：查看,可用,python,torch,pytorch,cuda,gpu
来源： https://www.cnblogs.com/yibeimingyue/p/16607547.html