Эта команда устанавливает переменную среды («CUDA_VISIBLE_DEVICES») для скрипта python перед запуском.
$ CUDA_VISIBLE_DEVICES=2,3 python my_script.py # Uses GPUs 2 and 3.
Он отлично работает на машине Linux, но в Windows он говорит, что
'CUDA_VISIBLE_DEVICES' is not recognized as an internal or external command
Можно ли сделать это в Windows без изменения скрипта Python?
вы пробовали export CUDA_VISIBLE_DEVICES=2,3?






Для одного лайнера просто запустите в Windows:
set CUDA_VISIBLE_DEVICES=2,3 & python my_script.py
Для окон у меня тоже это сработало, спасибо Адонису. у вас есть два варианта
в командной строке вы можете использовать следующую команду
set TF_CUDNN_USE_AUTOTUNE=0 CUDA_VISIBLE_DEVICES=1 & python demo\singleperson.py
в основном файле python поместите следующие строки, ниже импортируйте вещи
import os
os.environ["CUDA_DEVICE_ORDER"] = "PCI_BUS_ID"
os.environ["CUDA_VISIBLE_DEVICES"] = "1"
затем запустите файл python в командной строке
python demo\singleperson.py
А как насчет
set CUDA_VISIBLE_DEVICES=2,3 & python my_script.py?