diff --git a/.devcontainer/devcontainer.json b/.devcontainer/devcontainer.json
index 2f7dc4e..29fb222 100644
--- a/.devcontainer/devcontainer.json
+++ b/.devcontainer/devcontainer.json
@@ -85,7 +85,7 @@
 		// 	"pnpmVersion": "latest",
 		// 	"nvmVersion": "latest"
 		// },
-		"ghcr.io/devcontainers/features/nvidia-cuda:1": {
+		"ghcr.io/devcontainers/features/nvidia-cuda:3": {
 			"cudaVersion": "12.5",
 			"cudnnVersion": "8.9.5.29"
 		},