refine

.devcontainer/devcontainer.json

 // For format details, see https://aka.ms/devcontainer.json. For config options, see the
 // README at: https://github.com/devcontainers/templates/tree/main/src/docker-in-docker
 {
-	"name": "Docker in Docker",
+	"name": "Python and Docker-in-Docker with NVIDIA GPU Support",
 	// Or use a Dockerfile or Docker Compose file. More info: https://containers.dev/guide/dockerfile
-	"image": "mcr.microsoft.com/devcontainers/base:bullseye",
+	"image": "mcr.microsoft.com/devcontainers/python:1-3.12-bullseye",
 
 	"features": {
 		"ghcr.io/devcontainers/features/docker-in-docker:2": {
-			"version": "latest",
-			"enableNonRootDocker": "true",
-			"moby": "true"
-		}
-	}
+			"version": "latest"
+		},
+		"ghcr.io/devcontainers/features/nvidia-cuda:1": {
+			"cudaVersion": "12.2",
+			"installToolkit": true
+		},
+		"ghcr.io/devcontainers-extra/features/ruff:1": {},
+		"ghcr.io/va-h/devcontainers-features/uv:1": {}
+	},
+
+	"runArgs": ["--runtime", "nvidia", "--gpus", "all"],
 
+	// Enable GPU passthrough
+	"hostRequirements": {
+		"gpu": "optional"
+	}
 	// Use 'forwardPorts' to make a list of ports inside the container available locally.
 	// "forwardPorts": [],