“GPU de la lista de Pytorch” Código de respuesta

Pytorch obtiene número de GPU

torch.cuda.device_count()
Smoggy Squirrel

GPU de la lista de 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

Respuestas similares a “GPU de la lista de Pytorch”

Explore las respuestas de código populares por idioma

Explorar otros lenguajes de código