.vscode/launch.json

{
    "version": "0.2.0",
    "configurations": [
        {
            "name": "Python Debugger: main.py",
            "type": "debugpy",
            "request": "launch",
            "program": "main.py",
            "console": "integratedTerminal",
            "justMyCode": false
        },
        {
            "name": "Python: Debug Current File",
            "type": "debugpy",
            "request": "launch",
            "program": "${file}",
            "console": "integratedTerminal",
            "justMyCode": false,
            "args": []
        },
        {
            "name": "Python: pytest",
            "type": "debugpy",
            "request": "launch",
            "program": "./.venv/bin/pytest",
            "args": [
                "-s"
            ],
            "console": "integratedTerminal",
            "justMyCode": false,
        },
        {
            "name": "Python: torchrun",
            "type": "debugpy",
            "request": "launch",
            "program": "./.venv/bin/torchrun",
            "args": [
                "--nproc-per-node=gpu",
                "--master_port=5535", // change port
                "main.py"
            ],
            "console": "integratedTerminal",
            "justMyCode": false,
        }
    ],
}