Pytorch Получите номер графического процессора
torch.cuda.device_count()
Smoggy Squirrel
torch.cuda.device_count()