VScode中如何为subprocess启动的PyTorch代码设置断点调试？

方案1：重构代码绕开subprocess调用（最推荐，成本最低） #

大部分场景下使用subprocess调用内部脚本，本质是为了传递配置参数，完全可以直接引入目标脚本的入口函数执行，让所有逻辑跑在同一个进程内，断点自然生效。
示例改造代码：

# 假设你的cmd指向的是项目内的train.py脚本的入口
from train import main as train_entry

def main():
    args = parse_args()
    # 直接将解析好的配置参数传入目标入口函数，无需生成命令调用subprocess
    train_entry(args)

如果目标脚本的入口是从sys.argv读取参数，也可以手动构造argv列表传入再调用：

import sys
from train import main as train_entry

def main():
    args = parse_args()
    # 构造和cmd命令对应的参数列表
    sys.argv = ["train.py", "--lr", str(args.lr), "--batch-size", str(args.batch_size)]
    train_entry()

方案2：开启VSCode Python调试器的子进程追踪 #

如果不想修改原有代码逻辑，直接修改VSCode的调试配置即可让调试器自动捕获子进程的断点。
操作步骤：

1. 打开项目下的.vscode/launch.json文件，没有的话可以在调试面板点击「创建launch.json文件」选择Python环境生成
2. 在你使用的调试配置项中新增"subProcess": true配置，参考示例：

{
    "version": "0.2.0",
    "configurations": [
        {
            "name": "Python: 调试当前文件",
            "type": "python",
            "request": "launch",
            "program": "${file}",
            "console": "integratedTerminal",
            "subProcess": true, // 开启子进程调试
            "justMyCode": false // 如果需要调试第三方库代码可以添加这行，不需要可省略
        }
    ]
}

3. 保存配置后重启调试，子进程内的断点即可正常触发。

方案3：使用debugpy手动挂载调试（兼容复杂场景） #

如果上述方案不生效（比如子进程启动逻辑特殊、运行在隔离环境中等），可以通过debugpy手动给子进程挂载调试器：

• 先安装debugpy：pip install debugpy
• 修改construct_cmd逻辑，给生成的命令添加debugpy启动参数，示例：

def construct_cmd(args):
    # 原有命令生成逻辑不变，在开头添加debugpy相关参数
    base_cmd = f"python -m debugpy --listen 127.0.0.1:5678 --wait-for-client "
    # 拼接原有命令
    return base_cmd + original_cmd

上述参数会让子进程启动后等待调试器连接再执行代码

• 在launch.json中新增一个attach类型的调试配置：

{
    "name": "Python: Attach 子进程",
    "type": "python",
    "request": "attach",
    "connect": {
        "host": "127.0.0.1",
        "port": 5678
    }
}

• 先启动原有主进程调试，等子进程启动后，切换调试配置为「Python: Attach 子进程」启动，即可连接到子进程触发断点。

内容的提问来源于stack exchange，提问作者ofir1080