В более новых версиях pytorch вы можете установить устройство с помощью torch.device. Мне нужно использовать версию факела 0.3.1.post2 для некоторого устаревшего кода. Как настроить устройство для этой версии?






Насколько я знаю, вы можете использовать функцию set_device. Но это не поощряется. См. ссылка.
Предлагаемый метод заключается в том, что вы просто устанавливаете переменную окружения CUDA_VISIBLE_DEVICES. Вы можете запустить свой скрипт следующим образом.
CUDA_VISIBLE_DEVICES=GPU_ID python script_name.py
В вашей программе вы можете просто использовать .cuda() для использования графических процессоров. (например, model=model.cuda())