diff --git a/.devcontainer/Dockerfile b/.devcontainer/Dockerfile
index cc4327b..79e7f40 100644
--- a/.devcontainer/Dockerfile
+++ b/.devcontainer/Dockerfile
@@ -1,4 +1,4 @@
-FROM pytorch/pytorch:2.2.0-cuda12.1-cudnn8-devel
+FROM pytorch/pytorch:2.2.0-cuda12.1-cudnn8-runtime
 
 RUN <<EOF
 apt-get update