Update Dockerfile

Dockerfile

@@ -1,50 +1,44 @@
-FROM nvidia/cuda:11.8.0-base-ubuntu22.04
+FROM nvidia/cuda:11.3.1-base-ubuntu20.04
 
-# Remove any third-party apt sources to avoid issues with expiring keys.
-RUN rm -f /etc/apt/sources.list.d/*.list
+ENV DEBIAN_FRONTEND=noninteractive \
+    TZ=Europe/Paris
 
-# Install some basic utilities.
-RUN apt-get update && apt-get install -y \
+# Remove any third-party apt sources to avoid issues with expiring keys.
+# Install some basic utilities
+RUN rm -f /etc/apt/sources.list.d/*.list && \
+    apt-get update && apt-get install -y --no-install-recommends \
     curl \
     ca-certificates \
     sudo \
     git \
+    git-lfs \
+    zip \
+    unzip \
+    htop \
     bzip2 \
     libx11-6 \
+    build-essential \
+    libsndfile-dev \
+    software-properties-common \
     && rm -rf /var/lib/apt/lists/*
 
-# Create a working directory.
-RUN mkdir /app
-WORKDIR /app
-
-# Create a non-root user and switch to it.
-RUN adduser --disabled-password --gecos '' --shell /bin/bash user \
-    && chown -R user:user /app
-RUN echo "user ALL=(ALL) NOPASSWD:ALL" > /etc/sudoers.d/90-user
-USER user
-
-# All users can use /home/user as their home directory.
-ENV HOME=/home/user
-RUN mkdir $HOME/.cache $HOME/.config \
-    && chmod -R 777 $HOME
-
-# Download and install Micromamba.
-RUN curl -sL https://micro.mamba.pm/api/micromamba/linux-64/1.1.0 \
-    | sudo tar -xvj -C /usr/local bin/micromamba
-ENV MAMBA_EXE=/usr/local/bin/micromamba \
-    MAMBA_ROOT_PREFIX=/home/user/micromamba \
-    CONDA_PREFIX=/home/user/micromamba \
-    PATH=/home/user/micromamba/bin:$PATH
+# Install Python 3.10 and pip
+RUN add-apt-repository ppa:deadsnakes/ppa && \
+    apt-get update && apt-get install -y --no-install-recommends \
+    python3.10 \
+    python3.10-distutils \
+    && rm -rf /var/lib/apt/lists/* && \
+    update-alternatives --install /usr/bin/python3 python3 /usr/bin/python3.10 1 && \
+    update-alternatives --install /usr/bin/python python /usr/bin/python3.10 1 && \
+    curl https://bootstrap.pypa.io/get-pip.py -o get-pip.py && \
+    python get-pip.py && \
+    rm get-pip.py
 
-# Set up the base Conda environment by installing PyTorch and friends.
-COPY conda-linux-64.lock /app/conda-linux-64.lock
-RUN micromamba create -qy -n base -f /app/conda-linux-64.lock \
-    && rm /app/conda-linux-64.lock \
-    && micromamba shell init --shell=bash --prefix="$MAMBA_ROOT_PREFIX" \
-    && micromamba clean -qya
+# Set Python 3.10 as the default python version
+RUN update-alternatives --set python3 /usr/bin/python3.10 && \
+    update-alternatives --set python /usr/bin/python3.10
 
-# Fix for https://github.com/pytorch/pytorch/issues/97041
-RUN ln -s "$CONDA_PREFIX/lib/libnvrtc.so.11.8.89" "$CONDA_PREFIX/lib/libnvrtc.so"
+RUN apt-get update && apt-get install ffmpeg libsm6 libxext6  -y
 
 
 WORKDIR /code
@@ -69,4 +63,4 @@ WORKDIR $HOME/app
 # Copy the current directory contents into the container at $HOME/app setting the owner to the user
 COPY --chown=user . $HOME/app
 
-CMD ["python", "main.py"]
+CMD ["python", "main.py"]