fixed #144, closes #143

Author: Yi-01-ai

Dockerfile

@@ -3,27 +3,33 @@ ARG CUDA_VERSION="11.8.0"
 FROM mambaorg/micromamba:1.5.1 as micromamba
 FROM ${REGISTRY}/nvidia/cuda:${CUDA_VERSION}-devel-ubuntu22.04 as base
 
-#####
-# Setup user & common tools
-#####
-RUN apt update \
-  && apt install -y git ninja-build \
-  && rm -rf /var/lib/apt/lists/*
-
 #####
 # Setup micromamba
 #####
 
 USER root
 
 ARG MAMBA_USER=yi
-ARG MAMBA_USER_ID=56789
-ARG MAMBA_USER_GID=56789
+ARG MAMBA_USER_ID=1000
+ARG MAMBA_USER_GID=100
 ENV MAMBA_USER=$MAMBA_USER
 ENV MAMBA_ROOT_PREFIX="/opt/conda"
 ENV MAMBA_EXE="/bin/micromamba"
 ENV ENV_NAME=yi
 
+ENV DEBIAN_FRONTEND="noninteractive"
+ENV TZ="Asia/Shanghai"
+ENV LC_ALL=C.UTF-8
+ENV LANG=C.UTF-8
+
+RUN apt-get update -y \
+  && apt-get install -y sudo tzdata git ninja-build \
+  && useradd -ms /bin/bash -d /home/$MAMBA_USER $MAMBA_USER --uid $MAMBA_USER_ID --gid $MAMBA_USER_GID \
+  && usermod -aG sudo $MAMBA_USER \
+  && echo "$MAMBA_USER ALL=NOPASSWD: ALL" >> /etc/sudoers \
+  && rm -rf /var/lib/apt/lists/* \
+  && apt-get clean
+
 COPY --from=micromamba "$MAMBA_EXE" "$MAMBA_EXE"
 COPY --from=micromamba /usr/local/bin/_activate_current_env.sh /usr/local/bin/_activate_current_env.sh
 COPY --from=micromamba /usr/local/bin/_dockerfile_shell.sh /usr/local/bin/_dockerfile_shell.sh
@@ -41,9 +47,9 @@ CMD ["/bin/bash"]
 
 # Install dependencies
 
-WORKDIR /yi
-COPY ./conda-lock.yml .
+WORKDIR /home/${MAMBA_USER}/workspace/Yi
+COPY --chown=${MAMBA_USER_ID}:${MAMBA_USER_GID} ./conda-lock.yml .
 RUN micromamba create -y -n ${ENV_NAME} -f conda-lock.yml && \
   micromamba clean --all --yes
 
-COPY . .
+COPY --chown=${MAMBA_USER_ID}:${MAMBA_USER_GID} . .