diff --git a/.devcontainer/devcontainer.json b/.devcontainer/devcontainer.json
index b54530d..0ace8d4 100644
--- a/.devcontainer/devcontainer.json
+++ b/.devcontainer/devcontainer.json
@@ -30,5 +30,11 @@
     },
     "postCreateCommand": "uv sync --frozen && uv run scripts/download_assets.py",
     "postStartCommand": "uv run pre-commit install",
-    "remoteUser": "vscode"
+    "remoteUser": "vscode",
+    "runArgs": [
+        "--gpus",
+        "all",
+        "--runtime",
+        "nvidia"
+    ]
 }