py3

Dockerfile

@@ -1,19 +1,20 @@
-FROM nvidia/driver:418.40.04-ubuntu18.04
-RUN apt-get -y update \
-    && apt-get install -y software-properties-common \
-    && apt-get -y update \
-    && add-apt-repository universe
-RUN apt-get -y update
-RUN apt-get -y install python3
-RUN apt-get -y install python3-pip
+FROM nvidia/cuda:12.0.0-cudnn8-devel-ubuntu22.04
+
+# Using the Ubuntu image (our OS)
+# Update package manager (apt-get) 
+# and install (with the yes flag `-y`)
+# Python and Pip
+RUN apt-get update && apt-get install -y \
+    python3-pip \ 
+    wget 
 
 WORKDIR /code
 
 COPY ./requirements.txt /code/requirements.txt
 
-RUN python3-pip install --no-cache-dir --upgrade -r /code/requirements.txt
+RUN pip install --no-cache-dir --upgrade -r /code/requirements.txt
 
-RUN python3-pip install diffusers transformers accelerate scipy safetensors
+RUN pip install diffusers transformers accelerate scipy safetensors
 
 
 
@@ -33,7 +34,7 @@ WORKDIR $HOME/app
 COPY --chown=user . $HOME/app
 
 # Download and uzip truepic-size from Google Drive
-RUN python3-pip install gdown
+RUN pip install gdown
 RUN gdown --id 1MdXMu8xSuG8WClY3JSgtL3Vy1IWTmWZO
 RUN tar -xf truepic-sign-v0.1.0-ubuntu-latest.tar.gz
 RUN chmod +x truepic-sign