Update Dockerfile

Dockerfile

@@ -3,10 +3,16 @@
 FROM python:3.9
 
 RUN useradd -m -u 1000 user
-#USER user
+USER user
+
+
 ENV PATH="/home/user/.local/bin:/opt/conda/bin:$PATH"
+ENV HOME="/home/user"
+
+WORKDIR $HOME/app
+
 
-#WORKDIR /app
+RUN pip install --no-cache-dir --upgrade pip
 
 ARG PYTORCH_VERSION=2.1.0
 ARG PYTHON_VERSION=3.9 #8.10
@@ -18,59 +24,60 @@ ARG INSTALL_CHANNEL=pytorch
 # Automatically set by buildx
 ARG TARGETPLATFORM
 
-COPY ./train.sh /train.sh
-RUN chmod +x /train.sh
-
-USER root
+#USER root
 
 # Update basic dependencies we'll be using.
-RUN apt-get update && DEBIAN_FRONTEND=noninteractive apt-get install -y --no-install-recommends \
-        build-essential \
-        ca-certificates \
-        ccache \
-        curl \
-        python3 \
-        python3-pip \
-        git && \
-        rm -rf /var/lib/apt/lists/*
+#RUN apt-get update && DEBIAN_FRONTEND=noninteractive apt-get install -y --no-install-recommends \
+#        build-essential \
+#        ca-certificates \
+#        ccache \
+#        curl \
+#        python3 \
+#        python3-pip \
+#        git && \
+#        rm -rf /var/lib/apt/lists/*
 
 RUN git clone https://github.com/lee-ny/teaching_arithmetic.git /teaching_arithmetic 
 # && cd teaching_arithmetic && pip install -e .
 #RUN chmod +x /teaching_arithmetic
 # Install conda
 # translating Docker's TARGETPLATFORM into mamba arches
-RUN case ${TARGETPLATFORM} in \
-         "linux/arm64")  MAMBA_ARCH=aarch64  ;; \
-         *)              MAMBA_ARCH=x86_64   ;; \
-    esac && \
-    curl -fsSL -v -o ~/mambaforge.sh -O  "https://github.com/conda-forge/miniforge/releases/download/${MAMBA_VERSION}/Mambaforge-${MAMBA_VERSION}-Linux-${MAMBA_ARCH}.sh"
-RUN chmod +x ~/mambaforge.sh && \
-    bash ~/mambaforge.sh -b -p /opt/conda && \
-    rm ~/mambaforge.sh
+#RUN case ${TARGETPLATFORM} in \
+#         "linux/arm64")  MAMBA_ARCH=aarch64  ;; \
+#         *)              MAMBA_ARCH=x86_64   ;; \
+#    esac && \
+#    curl -fsSL -v -o ~/mambaforge.sh -O  "https://github.com/conda-forge/miniforge/releases/download/${MAMBA_VERSION}/Mambaforge-${MAMBA_VERSION}-Linux-${MAMBA_ARCH}.sh"
+#RUN chmod +x ~/mambaforge.sh && \
+#    bash ~/mambaforge.sh -b -p /opt/conda && \
+#    rm ~/mambaforge.sh
 
 # Install pytorch
 # On arm64 we exit with an error code
-RUN case ${TARGETPLATFORM} in \
-         "linux/arm64")  exit 1 ;; \
-         *)              /opt/conda/bin/conda update -y conda &&  \
-                         /opt/conda/bin/conda install -c "${INSTALL_CHANNEL}" -c "${CUDA_CHANNEL}" -y "python=${PYTHON_VERSION}" "pytorch=$PYTORCH_VERSION" "pytorch-cuda=$(echo $CUDA_VERSION | cut -d'.' -f 1-2)"  ;; \
-    esac && \
-    /opt/conda/bin/conda clean -ya
-
-USER 1000
+#RUN case ${TARGETPLATFORM} in \
+#         "linux/arm64")  exit 1 ;; \
+#         *)              /opt/conda/bin/conda update -y conda &&  \
+#                         /opt/conda/bin/conda install -c "${INSTALL_CHANNEL}" -c "${CUDA_CHANNEL}" -y "python=${PYTHON_VERSION}" "pytorch=$PYTORCH_VERSION" "pytorch-cuda=$(echo $CUDA_VERSION | cut -d'.' -f 1-2)"  ;; \
+#    esac && \
+#    /opt/conda/bin/conda clean -ya
+
+#USER 1000
 #COPY --chown=user ./requirements.txt requirements.txt
-COPY ./requirements.txt requirements.txt
-RUN pip install --no-cache-dir --upgrade -r requirements.txt
+#COPY ./requirements.txt requirements.txt
+#RUN pip install --no-cache-dir --upgrade -r requirements.txt
 
 #COPY --chown=user teaching_arithmetic /teaching_arithmetic
-#COPY --chown=user . /app
+COPY --chown=user . $HOME/app
+RUN pip install --no-cache-dir --upgrade -r requirements.txt
 #COPY . /app
-COPY ./out /out
+#COPY ./out /out
 #RUN chmod 777 -R /out
 
 #RUN chmod +x /out
-COPY ./out/addition_train /out/
+#COPY ./out/addition_train /out/
 #RUN chmod +x /teaching_arithmetic/train.py
+#COPY ./train.sh /train.sh
+#RUN chmod +x /train.sh
+
 
 # Expose the secret DEBUG at buildtime and use its value as git remote URL
 #RUN --mount=type=secret,id=DEBUG,mode=0444,required=true \