diff --git a/.devcontainer/devcontainer.json b/.devcontainer/devcontainer.json
index be1fa12..3e5c8d4 100644
--- a/.devcontainer/devcontainer.json
+++ b/.devcontainer/devcontainer.json
@@ -34,8 +34,7 @@
     },
     "remoteUser": "vscode",
     "runArgs": [
-        "--gpus",
-        "all"
+        "--device", "nvidia.com/gpu=all"
     ],
     // Ensure the .venv persists using a named volume for performance and parity
     "mounts": [