diff --git a/.devcontainer/gpu/devcontainer.json b/.devcontainer/gpu/devcontainer.json
index 27dd73b..7a2d03c 100644
--- a/.devcontainer/gpu/devcontainer.json
+++ b/.devcontainer/gpu/devcontainer.json
@@ -9,8 +9,7 @@
 		"--cap-add=SYS_PTRACE",
 		"--security-opt",
 		"seccomp=unconfined",
-		"--device",
-		"nvidia.com/gpu=all"
+		"--gpus=all"
 	],
 	// Set *default* container specific settings.json values on container create.
 	"customizations": {