-
Notifications
You must be signed in to change notification settings - Fork 13
Expand file tree
/
Copy pathdevcontainer.json
More file actions
38 lines (38 loc) · 1.08 KB
/
devcontainer.json
File metadata and controls
38 lines (38 loc) · 1.08 KB
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
{
"name": "Python 3.11",
"image": "mcr.microsoft.com/devcontainers/python:3.11",
"runArgs": [
"--runtime",
"nvidia",
"--gpus",
"all",
// optional but make sure CUDA workloads are available
"--env",
"NVIDIA_VISIBLE_DEVICES=all",
// optional but make sure CUDA workloads are available
"--env",
"NVIDIA_DRIVER_CAPABILITIES=compute,utility"
],
"customizations": {
"vscode": {
"extensions": [
"ms-python.black-formatter",
"ms-python.flake8",
"ms-python.isort",
"ms-python.vscode-pylance",
"ms-python.python",
"ms-python.debugpy",
"ms-python.vscode-python-envs",
"hbenl.vscode-test-explorer"
]
}
},
"hostRequirements": {
"gpu": "optional"
},
"remoteEnv": {
// optional but make sure CUDA workloads are available
"NVIDIA_VISIBLE_DEVICES": "all",
"NVIDIA_DRIVER_CAPABILITIES": "compute,utility"
}
}