此命令在运行之前为 python 脚本设置一个环境变量(“CUDA_VISIBLE_DEVICES”)。
$ CUDA_VISIBLE_DEVICES=2,3 python my_script.py # Uses GPUs 2 and 3.
Run Code Online (Sandbox Code Playgroud)
它在 Linux 机器上运行良好,但在 Windows 上,它说
'CUDA_VISIBLE_DEVICES' is not recognized as an internal or external command
Run Code Online (Sandbox Code Playgroud)
是否可以在不更改 python 脚本的情况下在 Windows 上执行此操作?
对于单班轮,只需在 Windows 上运行:
set CUDA_VISIBLE_DEVICES=2,3 & python my_script.py
Run Code Online (Sandbox Code Playgroud)
| 归档时间: |
|
| 查看次数: |
5514 次 |
| 最近记录: |