pytorch Obtenha o número da GPU

torch.cuda.device_count()
Smoggy Squirrel