diff --git a/Dockerfile b/Dockerfile
index a80afa2..45a7c58 100644
--- a/Dockerfile
+++ b/Dockerfile
@@ -1,4 +1,4 @@
-FROM nvidia/cuda:12.2.2-runtime-rockylinux8
+FROM nvidia/cuda:12.2.2-cudnn8-runtime-rockylinux8
 
 RUN dnf install python39 python39-pip openssh-server -y