Dev: update dockerfile (#76)

* update: env * update: env * update: dockerfile * update: numpy version
Vincentqyw · Sep 22, 2024 · 74888a4 · 74888a4
1 parent 066a865
commit 74888a4
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git 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
@@ -11,7 +11,7 @@ Jinja2
 kornia
 loguru
 matplotlib
-numpy
+numpy==1.23.5
 onnxruntime
 omegaconf
 opencv-python