Update Dockerfile

Dockerfile

@@ -1,44 +1,50 @@
-FROM nvidia/cuda:11.3.1-base-ubuntu20.04
-
-ENV DEBIAN_FRONTEND=noninteractive \
-    TZ=Europe/Paris
+FROM nvidia/cuda:11.8.0-base-ubuntu22.04
 
 # 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 \
+RUN rm -f /etc/apt/sources.list.d/*.list
+
+# Install some basic utilities.
+RUN apt-get update && apt-get install -y \
     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/*
-
-# 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 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
-
-RUN apt-get update && apt-get install ffmpeg libsm6 libxext6  -y
+    && 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
+
+# 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
+
+# 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"
 
 
 WORKDIR /code