vscode分布式训练debug(记录贴)

  • 有参数的debug

这个很容易做到,通过launch.json就可以解决。

  1. 选择下图第四个
    标题
  2. 新建launch.json
  3. 主要是添加args
    {
          "version": "0.2.0",
          "configurations": [
                {
                      "name": "Python: Current File"
                      "type": "python",
                      "request": "launch",
                      "program": "${file}",
                      "console": "integratedTerminal",
                      "args": [
                            "arg1", "xxx",
                            "arg2", "xxx",
                       ]
                }
    } 

  • 分布式训练debug
    这个相比只有参数的debug复杂了一点,但是也是只有launch.json的区别:
    {
        "version": "0.2.0",
        "configurations": [
    
              {
                    "name": "Python: Current File",
                    "type": "python",
                    "request": "launch",
                    "console": "integratedTerminal",
                    "args": [
                        "--nproc_per_node=2",
                        "main.py",
                    ],
                    "env": {"CUDA_VISIBLE_DEVICES":"0,1"},
        
                     "program": "/home/lucas/anaconda3/envs/py39-pt112-cu113/lib/python3.9/site-packages/torch/distributed/run.py",
              }
            ]
    } 

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值