pytorch查看GPU序号型号

最新推荐文章于 2024-09-30 19:40:46 发布

Enochzhu

最新推荐文章于 2024-09-30 19:40:46 发布

阅读量7.1k

点赞数 5

原文链接：https://blog.csdn.net/u013066730/article/details/85300897

版权

import torch
 
print(torch.cuda.is_available())
#cuda是否可用；
 
print(torch.cuda.device_count())
#返回gpu数量；
 
print(torch.cuda.get_device_name(0))
#返回gpu名字，设备索引默认从0开始；
 
print(torch.cuda.current_device())
#返回当前设备索引；