xref: /aosp_15_r20/external/pytorch/.devcontainer/cuda/devcontainer.json (revision da0073e96a02ea20f0ac840b70461e3646d07c45) 
1// For format details, see https://aka.ms/devcontainer.json. For config options, see the
2// README at: https://github.com/devcontainers/templates/tree/main/src/anaconda
3{
4  "name": "PyTorch - CUDA",
5  "build": {
6    "context": "../..",
7    "dockerfile": "../Dockerfile",
8    "args": {
9      "USERNAME": "vscode",
10      "BUILDKIT_INLINE_CACHE": "0",
11      "CUDA_VERSION": "11.8.0",
12      "CLANG_VERSION": ""
13    }
14  },
15  "runArgs": ["--gpus", "all"],
16// Use 'forwardPorts' to make a list of ports inside the container available locally.
17  // "forwardPorts": [],
18
19  // Use 'postCreateCommand' to run commands after the container is created.
20  "postCreateCommand": "bash .devcontainer/scripts/install-dev-tools.sh",
21
22  // Configure tool-specific properties.
23  // "customizations": {},
24  "customizations": {
25    "vscode": {
26      "extensions": ["streetsidesoftware.code-spell-checker"]
27    }
28  },
29
30  // Features to add to the dev container. More info: https://containers.dev/features.
31  "features": {
32    // This is needed for lintrunner
33    "ghcr.io/devcontainers/features/rust:1" : {}
34  }
35  // Uncomment to connect as root instead. More info: https://aka.ms/dev-containers-non-root.
36  // "remoteUser": "root"
37}