43 lines
1.5 KiB
JSON
43 lines
1.5 KiB
JSON
// For format details, see https://aka.ms/devcontainer.json. For config options, see the
|
||
// README at: https://github.com/devcontainers/templates/tree/main/src/anaconda
|
||
{
|
||
"name": "PyTorch - CUDA",
|
||
"build": {
|
||
"context": "./",
|
||
"dockerfile": "../Dockerfile",
|
||
"args": {
|
||
"USERNAME": "vscode",
|
||
"BUILDKIT_INLINE_CACHE": "0",
|
||
"CUDA_VERSION": "12.8.0",
|
||
"CLANG_VERSION": ""
|
||
}
|
||
},
|
||
"runArgs": ["--runtime", "nvidia", "--gpus", "all"],
|
||
// Mount the full repo only after the container starts
|
||
"workspaceMount": "source=${localWorkspaceFolder},target=/workspace/pytorch,type=bind,consistency=cached",
|
||
"workspaceFolder": "/workspace/pytorch",
|
||
"containerEnv": {
|
||
"PIP_USER": "0" // <‑‑ disable implicit --user
|
||
},
|
||
// Use 'forwardPorts' to make a list of ports inside the container available locally.
|
||
// "forwardPorts": [],
|
||
|
||
// Use 'postCreateCommand' to run commands after the container is created.
|
||
"postCreateCommand": "bash .devcontainer/scripts/install-dev-tools.sh",
|
||
|
||
// Configure tool-specific properties.
|
||
// "customizations": {},
|
||
"customizations": {
|
||
"vscode": {
|
||
"extensions": ["streetsidesoftware.code-spell-checker"]
|
||
}
|
||
},
|
||
|
||
// Features to add to the dev container. More info: https://containers.dev/features.
|
||
"features": {
|
||
// This is needed for lintrunner
|
||
"ghcr.io/devcontainers/features/rust:1" : {}
|
||
}
|
||
// Uncomment to connect as root instead. More info: https://aka.ms/dev-containers-non-root.
|
||
// "remoteUser": "root"
|
||
} |