Pytorch obtiene número de GPU

torch.cuda.device_count()
Smoggy Squirrel