VSCode中调试通过torchrun实现的分布式训练启动程序

train.sh文件实现torchrun如下

bash 复制代码
#!/bin/bash

py3clean ./
CUDA_VISIBLE_DEVICES=3 torchrun --nproc_per_node=1 --master_port=9006 tools/train.py \
                        configs/basicvsr_plusplus_vimeo90k_bd.py \
                        --seed 0 \

需要进行更改来DeBug,改成launch.json如下所示,怎么打开launch.json百度一下

bash 复制代码
{
    "version": "0.2.0",
    "configurations": [
        {
            "python":"${command:python.interpreterPath}",
            "name": "Debug Training Script",
            "type": "python",
            "request": "launch",
            "program": "tools/train.py",
            "args": [
                "--seed",
                "0",
                "configs/basicvsr_plusplus_vimeo90k_bd.py"            
            ],
            
            "cwd": "${workspaceFolder}",
            "env": {
                "CUDA_VISIBLE_DEVICES": "3"
            },
            "console": "integratedTerminal",
            "stopOnEntry": false,

            "justMyCode": false
        }
    ]
}

然后打开如下窗口,点击这个按钮开始调试

相关推荐
八月正好an1 天前
visual studio安装本地帮助手册
ide·visual studio
张较瘦_1 天前
环境搭建 | [入门级]VSCode(Cursor|Trae|Qoder)搭建Java(Springboot3)企业开发环境全流程
java·ide·vscode
生莫甲鲁浪戴1 天前
Android Studio新手开发第二十一天
android·ide·android studio
生莫甲鲁浪戴1 天前
Android Studio新手开发第二十二天
android·ide·android studio
天赐细莲1 天前
(Linux) WSL 通过 VSCode 连接不执行 profile 问题(登录Shell问题)
linux·运维·vscode
暮色驶过苍茫2 天前
VSCode 配置 SSH 远程连接
ide·vscode·ssh
lichong9512 天前
Android studio 修改包名
android·java·前端·ide·android studio·大前端·大前端++
微风粼粼2 天前
eclipse 导入javaweb项目,以及配置教程(傻瓜式教学)
java·ide·eclipse
fruge2 天前
Visual Studio 2022安装时共享组件、工具、SDK路径无法更改的问题
ide·visual studio
我狸才不是赔钱货2 天前
DevOps:打破开发与运维之间的高墙
运维·vscode·docker·devops