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

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

相关推荐
笑醉踏歌行5 小时前
idea应用代码配色网站
java·ide·intellij-idea
山顶望月川6 小时前
并行科技MaaS平台支持文心4.5系列开源模型调用
人工智能·机器学习·编辑器
韩初心7 小时前
使用 visual studio 2022 编译 Lua5.4.8
ide·visual studio·lua5.4
GDAL7 小时前
Node.js REPL 教程
node.js·编辑器·vim
myloveasuka8 小时前
信号操作集函数
linux·运维·服务器·c语言·c++·vscode
向宇it8 小时前
【unity游戏开发——网络】网络游戏通信方案——强联网游戏(Socket长连接)、 弱联网游戏(HTTP短连接)
网络·http·游戏·unity·c#·编辑器·游戏引擎
comeilmforever9 小时前
IDEA2025 Version Control 窗口 local changes显示
java·ide·intellij-idea
kiss strong13 小时前
好用的自带AI功能的国产IDE
ide
成遇10 天前
在Vscode中安装Sass并配置
vscode·rust·sass
切韵10 天前
Unity编辑器扩展:UI绑定复制工具
ui·unity·编辑器