.devcontainer.json

{
  "name": "hackathon",
  "image": "mcr.microsoft.com/devcontainers/python:3",
  "extensions": ["Iterative.dvc", "ms-python.python", "redhat.vscode-yaml"],
  "features": {
      "ghcr.io/iterative/features/dvc:1": {},
      "ghcr.io/devcontainers/features/nvidia-cuda:1": {
          "installCudnn": true
      }
  },
  "postCreateCommand": "pip3 install --user -r requirements.txt"
}