在调试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"
}
}
]
}
|