diff --git a/Dockerfile b/Dockerfile
index 8dc6460..e5460f0 100644
--- a/Dockerfile
+++ b/Dockerfile
@@ -1,4 +1,4 @@
-FROM pytorch/pytorch:2.4.1-cuda12.4-cudnn9-runtime
+FROM pytorch/pytorch:2.4.1-cuda12.4-cudnn9-devel
 LABEL maintainer="prime genesys"
 LABEL repository="genesys"