如何列出所有pytorch能用的GPU?pytorch查看GPU列表

Song • 2835 次浏览 • 0 个回复 • 2020年11月27日

你可以使用下面的方法列举出来可用的GPU列表:

>>> import torch
>>> available_gpus = [torch.cuda.device(i) for i in range(torch.cuda.device_count())]
>>> available_gpus
[<torch.cuda.device object at 0x7f2585882b50>]

# 获取可用数量
print(torch.cuda.device_count())

原创文章,转载请注明 :如何列出所有pytorch能用的GPU?pytorch查看GPU列表 - pytorch中文网
原文出处: https://ptorch.com/news/256.html
问题交流群 :168117787
提交评论
要回复文章请先登录注册
用户评论
  • 没有评论
Pytorch是什么?关于Pytorch! 【解决】PyTorch不再支持此GPU,因为它太旧了;源代码编译PyTorch