docker up

Dockerfile

@@ -1,25 +1,26 @@
-# CUDA-enabled base image with Python 3.11 support
+# CUDA-enabled base image with Ubuntu 20.04
 FROM nvidia/cuda:11.8.0-base-ubuntu20.04
 
 # Set environment variables
 ENV DEBIAN_FRONTEND=noninteractive
 ENV PYTHONUNBUFFERED=1
 
-# Install required system packages and Python 3.11
-RUN apt-get update && apt-get install -y \
+# Add deadsnakes PPA for Python 3.11
+RUN apt-get update && apt-get install -y software-properties-common && \
+    add-apt-repository -y ppa:deadsnakes/ppa && \
+    apt-get update
+
+# Install Python 3.11 and required packages
+RUN apt-get install -y \
     python3.11 python3.11-distutils python3.11-dev python3-pip git wget unzip && \
     rm -rf /var/lib/apt/lists/*
 
-# Update alternatives to ensure Python 3.11 is the default
+# Update alternatives to set Python 3.11 as default
 RUN update-alternatives --install /usr/bin/python3 python3 /usr/bin/python3.11 1 && \
     update-alternatives --config python3
 
-# **Important:** Make pip use Python 3.11
-RUN update-alternatives --install /usr/bin/pip pip /usr/bin/pip3.11 1 && \
-    update-alternatives --config pip 
-
 # Upgrade pip to the latest version
-RUN pip install --upgrade pip
+RUN python3 -m pip install --upgrade pip
 
 # Set the working directory
 WORKDIR /app
@@ -42,4 +43,4 @@ RUN pip install --no-cache-dir -r requirements.txt
 EXPOSE 7860
 
 # Start the Flask app
-CMD ["python3", "run.py"]
+CMD ["python3", "run.py"]