📌  相关文章
📜  pytorch list gpus - 汇编代码示例

📅  最后修改于: 2022-03-11 14:48:10.136000             🧑  作者: Mango

代码示例2
import torch
available_gpus = [torch.cuda.device(i) for i in range(torch.cuda.device_count())]
available_gpus
#[]