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
        }
    ]
}

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

相关推荐
小蕾Java9 小时前
PyCharm 2025:使用图文教程!
ide·python·pycharm
刘登辉10 小时前
idea使用联网缓存的pom进行离线开发
java·ide·intellij-idea·离线开发
小池先生10 小时前
用vscode查日志方便的搜索-复制所有匹配行功能
ide·vscode·编辑器
zstar-_15 小时前
FreeGIF:让制作动图变得超简单
编辑器·gif
Deryck_德瑞克15 小时前
IDEA编译时报错OOM的解决方案
java·ide·intellij-idea
JH307315 小时前
Idea中新建package包,变成了Directory
java·ide·intellij-idea
原小明16 小时前
【Conda】Conda虚拟环境配置系统环境变量,Jupter可使用
vscode·python·jupyter·conda
UpYoung!16 小时前
【Typora——MD编辑器】Typora最新 V1.12.1版:轻量级 Markdown 编辑器详细图文下载安装使用教程
学习·数学建模·编辑器·运维开发·个人开发
慧都小项18 小时前
Parasoft C/C++test如何将静态分析集成到VSCode中
vscode·静态分析·parasoft
猿小猴子20 小时前
主流 AI IDE 之一的 Qoder 和 Lingma IDE 介绍
ide·qoder·lingma