如何使用 pytorch 列出所有当前可用的 GPU?

How do I list all currently available GPUs with pytorch?(如何使用 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?的文章就介绍到这了，希望我们推荐的答案对大家有所帮助，也希望大家多多支持编程学习网！