v42

Dockerfile

@@ -1,4 +1,5 @@
-FROM nvidia/cuda:10.1-cudnn7-devel-ubuntu18.04
+FROM nvidia/cuda:11.8.0-cudnn8-devel-ubuntu20.04
+
 
 ENV DEBIAN_FRONTEND=noninteractive \
     PYTHONUNBUFFERED=1 \
@@ -6,14 +7,14 @@ ENV DEBIAN_FRONTEND=noninteractive \
     HOME=/home/user \
     PATH=/home/user/.local/bin:$PATH
 
-# Install Python 3.8 and git
+# Install Python 3.9 and git
 RUN apt-get update && \
-    apt-get install -y ffmpeg python3.7 python3.7-distutils python3.7-dev python3-pip git wget && \
+    apt-get install -y ffmpeg python3.9 python3.9-distutils python3-pip git && \
     rm -rf /var/lib/apt/lists/*
 
-# Make python3.8 default
-RUN update-alternatives --install /usr/bin/python python /usr/bin/python3.7 1 && \
-    update-alternatives --install /usr/bin/python3 python3 /usr/bin/python3.7 1
+# Make python3.9 default
+RUN update-alternatives --install /usr/bin/python python /usr/bin/python3.9 1 && \
+    update-alternatives --install /usr/bin/python3 python3 /usr/bin/python3.9 1
 
 # Create non-root user and fix permissions
 RUN useradd -m -u 1000 user && \
@@ -23,15 +24,12 @@ RUN useradd -m -u 1000 user && \
 USER user
 WORKDIR /home/user/app
 
-
 # Copy files and install only Gradio
 COPY --chown=user:user . /home/user/app
-RUN python3.7 -m pip install --user --no-cache-dir Cython==0.29.28
-RUN python3.7 -m pip install --user --no-cache-dir "setuptools<60.0"
-RUN python3.7 -m pip install --user --no-cache-dir numpy==1.21.6
-RUN python3.7 -m pip install --user --no-cache-dir gradio mxnet-cu101mkl
-RUN python3.7 -m pip install --user --no-cache-dir torch==1.6.0+cu101 torchvision==0.7.0+cu101 torchaudio -f https://download.pytorch.org/whl/torch_stable.html
-RUN python3.7 -m pip install --user --no-cache-dir onnx==1.9.0 onnxruntime-gpu==1.4.0 insightface==0.2.1 kornia==0.5.4 dill opencv-python scikit-image requests==2.25.1 wandb
+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
 
 COPY --chown=user:user . .
 RUN chmod +x download_models.sh