pytorch

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
}