magnetic-tiles-defect/.devcontainer/devcontainer.json

{
  "name": "magnetic-tiles-defect",
  "image": "mcr.microsoft.com/devcontainers/python:3.10",
  "features": {
    "ghcr.io/devcontainers/features/nvidia-cuda:1": {
      "installCudnn": true
    }
  },
  "extensions": [
    "Iterative.dvc",
    "ms-python.python",
    "redhat.vscode-yaml"
  ],
  "postCreateCommand": "pip install --user -r requirements.txt; python -m dvc pull; bash .devcontainer/nvtop_install.sh"
}