diff --git a/Dockerfile b/Dockerfile
index a8a036172..5aedc6fd8 100644
--- a/Dockerfile
+++ b/Dockerfile
@@ -4,8 +4,8 @@ FROM ${BASE_IMAGE}
 
 ARG ONEFLOW_PIP_INDEX
 ARG ONEFLOW_PACKAGE_NAME=oneflow
-RUN pip install -f ${ONEFLOW_PIP_INDEX} ${ONEFLOW_PACKAGE_NAME}
-RUN python3 -m pip install "torch" "transformers==4.27.1" "diffusers[torch]==0.19.3"
+RUN pip install -f ${ONEFLOW_PIP_INDEX} ${ONEFLOW_PACKAGE_NAME} "nvidia-cudnn-cu11>=8.9,<9.0"
+RUN python3 -m pip install "torch" "transformers==4.27.1" "diffusers[torch]==0.19.3" "huggingface-hub==0.23.2"
 ADD . /src/onediff
 RUN python3 -m pip install -e /src/onediff
 RUN python3 -m pip install -e /src/onediff/onediff_diffusers_extensions