You need to enable JavaScript to run this app.
最新活动
大模型
产品
解决方案
定价
生态与合作
支持与服务
开发者
了解我们

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

火山引擎 最新活动