vscode调试 指定命令行参数和环境变量

在调试Python程序的时候,经常遇到需要输入命令行参数的情形,在vscode中应该如何配置呢?以下为博主的一个实际案例。

需要调试的命令行

python3 examples/train.py -d ./kodim --epochs 300 -lr 1e-4 --batch-size 16 --cuda --save

环境变量:CUDA_VISIBLE=7

vscode中的配置

在“运行和调试”界面,直接选择Python File。然后具体配置文件如下:

{
    "configurations": [
        {
            "name": "Python: 当前文件",
            "type": "python",
            "request": "launch",

            # 这里写你要调试的Python文件,不指定则默认为当前文件(${file})
            "program": "./examples/train.py",       
            "console": "integratedTerminal",

            # 这里写需要的命令行参数
            "args": [ 
                "-d", "./kodim",
                "--epochs","300", 
                "-lr", "1e-4", 
                "--batch-size", "1",
                "--cuda", "--save",
                "--test-batch-size", "1"
            ],

            # 这里写需要配置的环境变量,不需要则不写
            "env": {
                "CUDA_VISIBLE_DEVICES": "7" 
            }
        }
    ]
}

你可能感兴趣的:(科学炼丹,vscode,python,ide)