diff --git a/.devcontainer/cuda/devcontainer.json b/.devcontainer/cuda/devcontainer.json
index 42a7d70..a1bf977 100644
--- a/.devcontainer/cuda/devcontainer.json
+++ b/.devcontainer/cuda/devcontainer.json
@@ -13,7 +13,7 @@
 			"autoPull": true,
 			"version": "latest"
 		},
-		"ghcr.io/devcontainers/features/nvidia-cuda:1": {
+		"ghcr.io/devcontainers/features/nvidia-cuda:3": {
 			"installCudnnDev": true,
 			"installCudnn": true,
 			"installNvtx": true,