v44

Dockerfile

@@ -1,5 +1,4 @@
-FROM nvidia/cuda:11.8.0-cudnn8-devel-ubuntu20.04
-
+FROM nvidia/cuda:11.2.2-cudnn8-devel-ubuntu20.04
 
 ENV DEBIAN_FRONTEND=noninteractive \
     PYTHONUNBUFFERED=1 \
@@ -28,8 +27,8 @@ WORKDIR /home/user/app
 COPY --chown=user:user . /home/user/app
 RUN pip install --user pydantic==2.8.2 gradio 
 
-RUN pip install --user torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu118 
-RUN pip install --user --no-cache-dir mxnet-cu112 onnxruntime-gpu==1.12 Cython requests==2.25.1 insightface==0.2.1 kornia==0.5.4 dill opencv-python onnx numpy scikit-image  wandb
+RUN pip install --user torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu113
+RUN pip install --user --no-cache-dir mxnet-cu112 onnxruntime-gpu==1.12 Cython requests==2.25.1 insightface==0.2.1 kornia==0.5.4 numpy<=2.0 dill opencv-python onnx scikit-image  wandb
 
 COPY --chown=user:user . .
 RUN chmod +x download_models.sh