devcontainer.json (758B)
1 { 2 "image": "mcr.microsoft.com/devcontainers/universal:2", 3 "runArgs": [ 4 "--gpus", 5 "all" 6 ], 7 "hostRequirements": { 8 "cpus": 4 9 }, 10 "waitFor": "onCreateCommand", 11 "customizations": { 12 "vscode": { 13 "extensions": [ 14 "ms-python.python", 15 "ms-toolsai.jupyter", 16 "ms-python.isort" 17 ] 18 } 19 }, 20 "features": { 21 "ghcr.io/devcontainers/features/nvidia-cuda:1": { 22 "installCudnn": true, 23 "installNvtx": true, 24 "cudaVersion": "11.7", 25 "cudnnVersion": "8.5.0.96" 26 } 27 }, 28 "postCreateCommand": "", 29 "updateContentCommand": "python3 -m pip install --user -r requirements.txt" 30 }