pytorch obtient le numéro de GPU

torch.cuda.device_count()
Smoggy Squirrel