устройство

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

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

Параметры

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