pytorch mendapatkan nomor GPU

torch.cuda.device_count()
Smoggy Squirrel