Что эквивалентно этой команде Linux в Windows cmd?

Эта команда устанавливает переменную среды («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?

А как насчет set CUDA_VISIBLE_DEVICES=2,3 & python my_script.py?

Adonis 09.04.2018 16:46

вы пробовали export CUDA_VISIBLE_DEVICES=2,3?

Stefan 09.04.2018 16:46
Почему в Python есть оператор "pass"?
Почему в Python есть оператор "pass"?
Оператор pass в Python - это простая концепция, которую могут быстро освоить даже новички без опыта программирования.
Некоторые методы, о которых вы не знали, что они существуют в Python
Некоторые методы, о которых вы не знали, что они существуют в Python
Python - самый известный и самый простой в изучении язык в наши дни. Имея широкий спектр применения в области машинного обучения, Data Science,...
Основы Python Часть I
Основы Python Часть I
Вы когда-нибудь задумывались, почему в программах на Python вы видите приведенный ниже код?
LeetCode - 1579. Удаление максимального числа ребер для сохранения полной проходимости графа
LeetCode - 1579. Удаление максимального числа ребер для сохранения полной проходимости графа
Алиса и Боб имеют неориентированный граф из n узлов и трех типов ребер:
Оптимизация кода с помощью тернарного оператора Python
Оптимизация кода с помощью тернарного оператора Python
И последнее, что мы хотели бы показать вам, прежде чем двигаться дальше, это
Советы по эффективной веб-разработке с помощью Python
Советы по эффективной веб-разработке с помощью Python
Как веб-разработчик, Python может стать мощным инструментом для создания эффективных и масштабируемых веб-приложений.
2
2
4 818
2
Перейти к ответу Данный вопрос помечен как решенный

Ответы 2

Ответ принят как подходящий

Для одного лайнера просто запустите в Windows:

set CUDA_VISIBLE_DEVICES=2,3 & python my_script.py

Для окон у меня тоже это сработало, спасибо Адонису. у вас есть два варианта

  1. в командной строке вы можете использовать следующую команду

    set TF_CUDNN_USE_AUTOTUNE=0 CUDA_VISIBLE_DEVICES=1 & python demo\singleperson.py

  2. в основном файле python поместите следующие строки, ниже импортируйте вещи

     import os
    
     os.environ["CUDA_DEVICE_ORDER"] = "PCI_BUS_ID"
    
     os.environ["CUDA_VISIBLE_DEVICES"] = "1"
    

затем запустите файл python в командной строке

python demo\singleperson.py

Другие вопросы по теме