Windows cmd上的这个Linux命令相当于什么

Question

此命令在运行之前为 python 脚本设置环境变量 ("CUDA_VISIBLE_DEVICES"(。

$ 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

是否可以在不更改python脚本的情况下在Windows上执行此操作？

Answer 1

对于一个班轮，只需在Windows上运行：

set CUDA_VISIBLE_DEVICES=2,3 & python my_script.py

Answer 2

对于窗户，这也对我有用，这要归功于阿多尼斯。 您有两个选择

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 demosingleperson.py