本文介绍了如何列出所有带有pytorch的GPU?的处理方法，对大家解决问题具有一定的参考价值，需要的朋友们下面随着小编来一起学习吧！
问题描述
我知道我可以使用torch.cuda.current_device()访问当前的GPU，但是如何获取所有当前可用GPU的列表?
I know I can access the current GPU using torch.cuda.current_device(), but how can I get a list of all the currently available GPUs?
推荐答案
您可以通过执行以下操作列出所有可用的GPU:
You can list all the available GPUs by doing:
>>> import torch
>>> available_gpus = [torch.cuda.device(i) for i in range(torch.cuda.device_count())]
>>> available_gpus
[<torch.cuda.device object at 0x7f2585882b50>]
这篇关于如何列出所有带有pytorch的GPU?的文章就介绍到这了，希望我们推荐的答案对大家有所帮助，也希望大家多多支持！