“GPUS de liste Pytorch” Réponses codées

pytorch obtient le numéro de GPU

torch.cuda.device_count()
Smoggy Squirrel

GPUS de liste Pytorch

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

Réponses similaires à “GPUS de liste Pytorch”

Parcourir les réponses de code populaires par langue

Parcourir d'autres langages de code