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

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

相关推荐
QL.ql3 小时前
编译器的相关知识(入门时著)
编辑器
阿洛学长4 小时前
Visual Studio下载安装教程及其旧版本
ide·visual studio
海天一色y7 小时前
Pycharm(二十一)递归删除文件夹
ide·python·pycharm
嫣语岁月8 小时前
【BMS电池管理】基于BQ76920与STM32的BMS设计开发
c语言·vscode·stm32·单片机·嵌入式硬件
BillKu13 小时前
推荐 Eclipse Temurin 的 OpenJDK
java·ide·eclipse
非门由也14 小时前
Android studio安装教程——超详细(含安装包安装教程)
android·ide·android studio
996终结者17 小时前
同类软件对比(四):Jupyter vs PyCharm vs VS Code:Python开发工具终极选择指南
vscode·python·jupyter·pycharm·visual studio code
万粉变现经纪人1 天前
如何解决pip安装报错ModuleNotFoundError: No module named ‘python-dateutil’问题
开发语言·ide·python·pycharm·pandas·pip·httpx
Adorable老犀牛1 天前
可遇不可求的自动化运维工具 | 2 | 实施阶段一:基础准备
运维·git·vscode·python·node.js·自动化
Lecxcy_Kastreain1 天前
解决VSCode默认F5配置无法启动调试器的问题
ide·vscode·编辑器