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

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

相关推荐
GDAL2 小时前
vscode 插件开发activityba
vscode·编辑器
learn_coder2 小时前
在vscode中和obsidian中使用Mermaid
ide·vscode·编辑器
程序员的世界你不懂6 小时前
IDE 关联 Git 操作
ide·git
karry01307 小时前
高并发导致重复key问题--org.springframework.dao.DuplicateKeyException
java·数据库·ide
前端小超人rui17 小时前
UEditor 对接 秀米 手机编辑器流程与问题
编辑器
l1x1n018 小时前
Vim 编辑器常用操作详解(新手快速上手指南)
linux·编辑器·vim
合作小小程序员小小店21 小时前
web网页,在线%食谱推荐系统%分析系统demo,基于vscode,uniapp,vue,java,jdk,springboot,mysql数据库
vue.js·spring boot·vscode·spring·uni-app
牧天白衣.1 天前
快捷键——VsCode
vscode
蜕变的土豆1 天前
Visual Studio 旧版软件下载教程
ide·visual studio
Shan12051 天前
编辑器Vim的快速入门
linux·编辑器·vim