У меня два gpus, и я хочу открыть второй блокнот juptyer и убедиться, что все внутри него работает только на втором gpu, а не на первом. В идеале я хотел бы сделать это, запустив ячейку в начале, а не передавая устройство = 1 в нескольких местах.
Ответ в этом сценарии заключается в вызове установленного устройства.
import torch torch.cuda.set_device(1)
Документы, однако, не одобряют его. Согласно примечанию github о регистрации (https://github.com/pytorch/pytorch/issues/260) мой сценарий является единственным разумным вариантом использования. Предпочтительным способом является установка переменной среды, что также возможно в случае с ноутбуком:
import os os.environ['CUDA_VISIBLE_DEVICES']='1'