diff --git a/.devcontainer/devcontainer.json b/.devcontainer/devcontainer.json
index 9ca74df..22d9728 100644
--- a/.devcontainer/devcontainer.json
+++ b/.devcontainer/devcontainer.json
@@ -1,29 +1,22 @@
-// For format details, see https://aka.ms/devcontainer.json. For config options, see the
-// README at: https://github.com/devcontainers/templates/tree/main/src/ubuntu
+// For format details, see https://aka.ms/devcontainer.json.
 {
 	"name": "Ubuntu",
-	// Or use a Dockerfile or Docker Compose file. More info: https://containers.dev/guide/dockerfile
 	"image": "mcr.microsoft.com/devcontainers/base:noble",
 
 	// Features to add to the dev container. More info: https://containers.dev/features.
 	"features": {
+		// CUDA support with GPU passthrough
 		"ghcr.io/devcontainers/features/nvidia-cuda:2": {},
-		"ghcr.io/devcontainers/features/docker-outside-of-docker:1": {},
-		"ghcr.io/devcontainers/features/python:1"
+
+		// Python
+		"ghcr.io/devcontainers/features/python:1": {
+			"version": "3.10" // sync with .python-version file
+		},
+
+		// (Optional) access to system's docker host
+		"ghcr.io/devcontainers/features/docker-outside-of-docker:1": {}
 	},
 
-	// Needed for cuda in the container
+	// Needed for GPU passthrough for CUDA in the container
     "runArgs": ["--gpus=all"]
-
-	// Use 'forwardPorts' to make a list of ports inside the container available locally.
-	// "forwardPorts": [],
-
-	// Use 'postCreateCommand' to run commands after the container is created.
-	// "postCreateCommand": "uname -a",
-
-	// Configure tool-specific properties.
-	// "customizations": {},
-
-	// Uncomment to connect as root instead. More info: https://aka.ms/dev-containers-non-root.
-	// "remoteUser": "root"
 }