.vscode/launch.json

{
    // 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": "3d_grand_demo",
            "type": "python",
            "request": "launch",
            "module": "app",
            "justMyCode": false,
            "env": {
                "CUDA_VISIBLE_DEVICES": "1",
            },
            "args": []
        }
    ]
}