Могу ли я временно отключить CUDA в PyTorch? [дубликат] ⇐ Python

[Anonymous]

Я пытаюсь определить, оправданы ли затраты на графический процессор для конкретной задачи. В настоящее время у меня есть код PyTorch, работающий на машине с поддержкой графического процессора. Мне бы хотелось иметь возможность отключить графический процессор для одного запуска, чтобы провести прямое сравнение (при прочих равных условиях) преимуществ графического процессора для задачи.
Можно ли как-нибудь сказать? PyTorch игнорировать графический процессор или заставить ОС притвориться, что его нет? Я запускаю это на Ubuntu 20.04.1 LTS, используя Python 3.8.5 и Torch 1.7.1.
ОБНОВЛЕНИЕ: это код, который я унаследовал, поэтому я не могу говорить о его эффективности, но есть во многих местах кода есть такие вещи, как ...
tensors = get_tensors()
if torch.cuda.device_count() >= 1:
tensers = tensors.cuda()

... а для модели это...
num_cuda = torch.cuda.device_count()
model = get_model()
if num_cuda == 1:
model = model.cuda()
elif num_cuda > 1:
model = torch.nn.DataParallel(model)


Подробнее здесь: https://stackoverflow.com/questions/665 ... in-pytorch