diff --git a/.devcontainer/devcontainer.json b/.devcontainer/devcontainer.json
index 78558f8..3bf52d2 100644
--- a/.devcontainer/devcontainer.json
+++ b/.devcontainer/devcontainer.json
@@ -11,6 +11,8 @@
 		// Everythong else gets installed automatically by uv
 	},
 
-	// Needed for GPU passthrough for CUDA in the container
-    "runArgs": ["--gpus=all"]
+	// Pass-through the GPU for CUDA if one is available
+	"hostRequirements": {
+		"gpu": "optional" 
+	}
 }