मुझे पता है कि मैं torch.cuda.current_device() का उपयोग करके वर्तमान GPU तक पहुंच सकता हूं, लेकिन मैं वर्तमान में उपलब्ध सभी GPU की सूची कैसे प्राप्त कर सकता हूं?

2
vinzee 11 नवम्बर 2020, 00:10

1 उत्तर

सबसे बढ़िया उत्तर

आप सभी उपलब्ध 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>]
3
vinzee 10 नवम्बर 2020, 21:10