一、修改launch.json(VSCode)
{
// Use IntelliSense to learn about possible attributes.
// Hover to view descriptions of existing attributes.
// For more information, visit: https://go.microsoft.com/fwlink/?linkid=830387
"version": "0.2.0",
"configurations": [
{
"name": "Python: Current File",
"type": "python",
"request": "launch",
"program": "${file}",
"console": "integratedTerminal",
"justMyCode": true
}
]
}
注销掉代码:
// "program": "${file}",
// "console": "integratedTerminal",
添加代码:
"connect": {
"host": "localhost",
"port": 50678
}
{
// 使用 IntelliSense 了解相关属性。
// 悬停以查看现有属性的描述。
// 欲了解更多信息,请访问: https://go.microsoft.com/fwlink/?linkid=830387
"version": "0.2.0",
"configurations": [
{
"name": "Python: Current File",
"type": "debugpy",
"request": "attach",
// "program": "${file}",
// "console": "integratedTerminal",
"justMyCode": false,
"connect": {
"host": "localhost",
"port": 50678
}
}
]
}
二、train.sh
利用debugpy调试,设定调试用的GPU标号为2
#!/usr/bin/env bash
#修改gpu编号
export CUDA_VISIBLE_DEVICES=2,3
python3 -m debugpy --listen 50678 --wait-for-client train.py
三、在vscode的terminal运行:
sh -x train.sh
运行完成shell脚本后停在python脚本调用的入口,点击F5进入python调试。