Update Dockerfile

Dockerfile

@@ -1,21 +1,7 @@
-FROM nvidia/cuda:11.7.1-cudnn8-devel-ubuntu22.04
+FROM nvidia/cuda:11.3.1-cudnn8-devel-ubuntu18.04
+CMD nvidia-smi
 
 ENV DEBIAN_FRONTEND noninteractive
-
-# Install Python 3.9 and required packages
-RUN apt-get update && \
-    apt-get install -y python3.9 python3.9-dev python3-pip && \
-    apt-get clean && \
-    rm -rf /var/lib/apt/lists/*
-
-# Set Python 3.9 as the default
-RUN update-alternatives --install /usr/bin/python3 python3 /usr/bin/python3.9 1
-
-# Install pip for Python 3.9
-RUN wget https://bootstrap.pypa.io/get-pip.py && \
-    python3.9 get-pip.py && \
-    rm get-pip.py
-
 # Set up a new user named "user" with user ID 1000
 RUN useradd -m -u 1000 user
 
@@ -33,13 +19,13 @@ WORKDIR $HOME/app
 RUN git clone -b dev https://github.com/camenduru/DiffBIR $HOME/app
 
 # Install PyTorch and torchvision with a specific CUDA toolkit version
-RUN pip3 install torch==1.12.1+cu113 torchvision==0.13.1+cu113 -f https://download.pytorch.org/whl/torch_stable.html
+RUN pip install torch==1.12.1+cu113 torchvision==0.13.1+cu113 -f https://download.pytorch.org/whl/torch_stable.html
 
 # Install Python dependencies
-RUN pip3 install -q einops pytorch_lightning gradio omegaconf xformers==0.0.20 transformers lpips opencv-python
+RUN pip install -q einops pytorch_lightning gradio omegaconf xformers==0.0.20 transformers lpips opencv-python
 
 # Install open_clip from GitHub
-RUN pip3 install -q git+https://github.com/mlfoundations/[email protected]
+RUN pip install -q git+https://github.com/mlfoundations/[email protected]
 
 # Use sudo to install aria2 with elevated privileges
 USER root