vscode调试分布式代码

命令行：

CUDA_VISIBLE_DEVICES=0 python -m torch.distributed.launch --nproc_per_node 1 --master_port 12345 main.py --eval --cfg configs/swin_base_patch4_window7_224_22kto1k_finetune.yaml --resume pretrained/swin_base_patch4_window7_224_22kto1k.pth --data-path /home/teamlq/workspace/wangjian/wangjian_dataset/Imagenet/

vscode调试

修改launch.json文件如下：

{

    // 使用 IntelliSense 了解相关属性。

    // 悬停以查看现有属性的描述。

    // 欲了解更多信息，请访问: https://go.microsoft.com/fwlink/?linkid=830387

    "version": "0.2.0",

    "configurations": [

        {

            "name": "Python: 当前文件",

            "type": "python",

            "request": "launch",

            "program": "/home/teamlq/.conda/envs/swintransformer/lib/python3.8/site-packages/torch/distributed/launch.py",

            "console": "integratedTerminal",

            "justMyCode": true,

            "args": [

                "--nproc_per_node",

                "1",

                "--master_port",

                "12345",

                "main.py",

                "--eval",

                "--cfg",

                "configs/swin_base_patch4_window7_224_22kto1k_finetune.yaml",

                "--resume",

                "pretrained/swin_base_patch4_window7_224_22kto1k.pth",

                "--data-path",

                "/home/teamlq/workspace/wangjian/wangjian_dataset/Imagenet/"

            ],

            "env": {"CUDA_VISIBLE_DEVICES":"5"},

        }

    ]

}