pytorch环境下查看gpu是否可用


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

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

 

原创文章,作者:ItWorker,如若转载,请注明出处:https://blog.ytso.com/281238.html

(0)
上一篇 2022年8月20日
下一篇 2022年8月20日

相关推荐

发表回复

登录后才能评论