pytorch
37 строк · 1.2 Кб
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}