diff --git a/cuda/Dockerfile b/cuda/Dockerfile
index 64fdb79..db74f70 100644
--- a/cuda/Dockerfile
+++ b/cuda/Dockerfile
@@ -1,4 +1,4 @@
-FROM docker.io/nvidia/cuda:11.8.0-devel-ubuntu22.04
+FROM docker.io/nvidia/cuda:11.8.0-cudnn8-devel-ubuntu22.04
 
 ARG APT_MIRROR=https://mirrors.bfsu.edu.cn
 ARG ROOT_PASSWD=miao