Update Dockerfile

Dockerfile

@@ -1,8 +1,25 @@
 FROM nvidia/cuda:12.2.0-devel-ubuntu22.04
 
-RUN useradd -m -u 1000 user
+ARG PYTORCH_VERSION=2.4.0
+ARG PYTHON_VERSION=3.9
+ARG CUDA_VERSION=12.1
+ARG MAMBA_VERSION=24.3.0-0
+ARG CUDA_CHANNEL=nvidia
+ARG INSTALL_CHANNEL=pytorch
+# Automatically set by buildx
+ARG TARGETPLATFORM
+
+
+#ENV HOME=/home/user \
+#	PATH=/home/user/.local/bin:/opt/conda/bin:$PATH
+
 ENV PATH=/opt/conda/bin:$PATH
 
+RUN mkdir -p .cache
+#RUN mkdir -p data
+# I'm not sure how to allow later python files used here to write to .cache without making it world-writeable.
+RUN chmod 777 -R .cache
+#RUN chmod 777 -R data
 
 RUN apt-get update && DEBIAN_FRONTEND=noninteractive apt-get install -y --no-install-recommends \
         build-essential \
@@ -34,10 +51,9 @@ RUN case ${TARGETPLATFORM} in \
     esac && \
     /opt/conda/bin/conda clean -ya
 
-
 COPY ./requirements.txt requirements.txt
+RUN pip install -r requirements.txt
 
-RUN pip install --no-cache-dir --upgrade -r requirements.txt
 RUN git clone -b energy_star_dev https://github.com/huggingface/optimum-benchmark.git /optimum-benchmark && cd optimum-benchmark && pip install -e .
 
 USER user