Update Dockerfile

Dockerfile

@@ -1,99 +1,91 @@
-FROM nvidia/cuda:12.5.1-cudnn-devel-ubuntu20.04
-
 ENV DEBIAN_FRONTEND=noninteractive \
-    TZ=Europe/Paris \
-    HOME=/root
+	TZ=Europe/Paris
 
-# Remove any third-party apt sources and install basic utilities
+# 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 \
-      wget \
-      procps \
-      git-lfs \
-      zip \
-      aria2 \
-      unzip \
-      ffmpeg \
-      htop \
-      vim \
-      nano \
-      bzip2 \
-      libx11-6 \
-      build-essential \
-      libsndfile-dev \
-      software-properties-common \
+    curl \
+    ca-certificates \
+    sudo \
+    git \
+    wget \
+    procps \
+    git-lfs \
+    zip \
+    aria2 \
+    unzip \
+    ffmpeg \
+    htop \
+    vim \
+    nano \
+    bzip2 \
+    libx11-6 \
+    build-essential \
+    libsndfile-dev \
+    software-properties-common \
  && rm -rf /var/lib/apt/lists/*
 
-# Install nvtop
 RUN add-apt-repository ppa:flexiondotorg/nvtop && \
     apt-get upgrade -y && \
     apt-get install -y --no-install-recommends nvtop
 
-# Install Node.js and configurable-http-proxy
-RUN curl -sL https://deb.nodesource.com/setup_21.x | bash - && \
+RUN curl -sL https://deb.nodesource.com/setup_21.x  | bash - && \
     apt-get install -y nodejs && \
     npm install -g configurable-http-proxy
 
-# Set working directory
+# Create a working directory
 WORKDIR /app
 
-# Install Conda in /root/miniconda and update PATH
+# Create a user account but continue using root
+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
+
+# Create home directories that will be accessible
+ENV HOME=/home/user
+RUN mkdir -p $HOME/.cache $HOME/.config \
+ && chmod -R 777 $HOME
+
+# Set up the Conda environment (as root)
 ENV CONDA_AUTO_UPDATE_CONDA=false \
-    PATH=/root/miniconda/bin:$PATH
-RUN curl -sLo /tmp/miniconda.sh https://repo.continuum.io/miniconda/Miniconda3-py39_4.10.3-Linux-x86_64.sh && \
-    chmod +x /tmp/miniconda.sh && \
-    /tmp/miniconda.sh -b -p /root/miniconda && \
-    rm /tmp/miniconda.sh && \
-    conda clean -ya
+    PATH=/opt/conda/bin:$PATH
+RUN curl -sLo /tmp/miniconda.sh https://repo.continuum.io/miniconda/Miniconda3-py39_4.10.3-Linux-x86_64.sh \
+ && chmod +x /tmp/miniconda.sh \
+ && /tmp/miniconda.sh -b -p /opt/conda \
+ && rm /tmp/miniconda.sh \
+ && conda clean -ya
 
 WORKDIR /app
 
-#######################################
-# Additional root-level operations
-#######################################
-
-# Install extra Debian packages (packages.txt must be provided)
+# User Debian packages
 RUN --mount=target=/root/packages.txt,source=packages.txt \
     apt-get update && \
-    xargs -r -a /root/packages.txt apt-get install -y --no-install-recommends && \
-    rm -rf /var/lib/apt/lists/*
+    xargs -r -a /root/packages.txt apt-get install -y --no-install-recommends \
+    && rm -rf /var/lib/apt/lists/*
 
-# Run startup script (if provided)
 RUN --mount=target=/root/on_startup.sh,source=on_startup.sh,readwrite \
-    bash /root/on_startup.sh
-
-# Create data directory
-RUN mkdir /data
+	bash /root/on_startup.sh
 
-#######################################
-# Python packages and application files
-#######################################
+RUN mkdir -p /data && chown user:user /data
 
-# Install Python packages from requirements.txt (ensure jupyterlab is included here or install explicitly)
+# Python packages
 RUN --mount=target=requirements.txt,source=requirements.txt \
     pip install --no-cache-dir --upgrade -r requirements.txt
 
-# Explicitly install JupyterLab if not already in requirements.txt
-RUN pip install --no-cache-dir jupyterlab
-
-# Copy the current directory contents into /app
+# Copy the current directory contents into the container
 COPY . /app
 
 RUN chmod +x start_server.sh
 
-# Overwrite Jupyter's login template if needed
-COPY login.html /root/miniconda/lib/python3.9/site-packages/jupyter_server/templates/login.html
+COPY login.html /opt/conda/lib/python3.9/site-packages/jupyter_server/templates/login.html
 
 ENV PYTHONUNBUFFERED=1 \
-    GRADIO_ALLOW_FLAGGING=never \
-    GRADIO_NUM_PORTS=1 \
-    GRADIO_SERVER_NAME=0.0.0.0 \
-    GRADIO_THEME=huggingface \
-    SYSTEM=spaces \
-    SHELL=/bin/bash
+	GRADIO_ALLOW_FLAGGING=never \
+	GRADIO_NUM_PORTS=1 \
+	GRADIO_SERVER_NAME=0.0.0.0 \
+	GRADIO_THEME=huggingface \
+	SYSTEM=spaces \
+	SHELL=/bin/bash
 
 CMD ["./start_server.sh"]