Linux设置CUDA_VISIBLE_DEVICES运行Python脚本的Windows cmd等效命令
在Windows中无需修改Python脚本设置CUDA_VISIBLE_DEVICES的方法
当然可以实现!Windows和Linux的环境变量设置语法不同,但完全不用修改Python脚本就能达到同样的效果,分两种常用终端场景给你说明:
1. 命令提示符(CMD)
在CMD里,使用set命令临时设置环境变量,再串联运行Python脚本:
set CUDA_VISIBLE_DEVICES=2,3 && python my_script.py
set是Windows CMD中设置临时环境变量的命令,这个变量仅在当前终端会话生效,关闭CMD后就会失效&&用来确保先完成环境变量设置,再执行Python脚本
2. PowerShell
PowerShell的环境变量语法和CMD不同,需要用$env:前缀来设置:
$env:CUDA_VISIBLE_DEVICES=2,3; python my_script.py
这里的环境变量同样是临时的,只在当前PowerShell会话中有效,不会影响其他终端或系统全局设置
如果需要永久设置这个环境变量(不推荐,除非你所有脚本都固定用这几个GPU),可以通过「系统属性 → 高级 → 环境变量」添加用户或系统级的CUDA_VISIBLE_DEVICES变量,但临时设置的方式更灵活,不会影响其他程序的GPU使用。
内容的提问来源于stack exchange,提问作者TSR




