устройство

class torch.cuda.device(device) [source]

Контекстный менеджер, который изменяет выбранное устройство.

Параметры:

device (torch.device или int) – индекс устройства для выбора. Это ничто, если этот аргумент является отрицательным целым числом или None.