diff --git a/Dockerfile b/Dockerfile
index 863cdf8..7455862 100644
--- a/Dockerfile
+++ b/Dockerfile
@@ -1,5 +1,5 @@
 # Use an official conda-based Python image as a parent image
-FROM pytorch/pytorch:2.2.1-cuda12.1-cudnn8-runtime
+FROM pytorch/pytorch:2.4.0-cuda12.1-cudnn9-runtime
 LABEL maintainer vincentqyw
 ARG PYTHON_VERSION=3.10.10
 
diff --git a/requirements.txt b/requirements.txt
index e979c2f..4ec3af7 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -11,7 +11,7 @@ Jinja2
 kornia
 loguru
 matplotlib
-numpy
+numpy==1.23.5
 onnxruntime
 omegaconf
 opencv-python