Update Dockerfile

Dockerfile

@@ -27,11 +27,6 @@ RUN apt-get update && DEBIAN_FRONTEND=noninteractive apt-get install -y --no-ins
 # RUN useradd -m -u 1000 user
 # USER user
 
-# RUN pip install torch==1.9.1+cu111 torchvision==0.10.1+cu111 torchaudio==0.9.1 -f https://download.pytorch.org/whl/torch_stable.html
-RUN pip install torch==2.0.0 torchvision==0.15.1 torchaudio==2.0.1 --index-url https://download.pytorch.org/whl/cu118
-
-# COPY --chown=user . /home/user
-
 ARG USER_NAME=user
 ARG GROUP_NAME=berf
 ARG UID=1000
@@ -41,9 +36,14 @@ RUN groupadd -g ${GID} ${GROUP_NAME} \
     && useradd -ms /bin/sh -u ${UID} -g ${GID} ${USER_NAME}
 USER user
 
+
+# RUN pip install torch==1.9.1+cu111 torchvision==0.10.1+cu111 torchaudio==0.9.1 -f https://download.pytorch.org/whl/torch_stable.html
+RUN pip install torch==2.0.0 torchvision==0.15.1 torchaudio==2.0.1 --index-url https://download.pytorch.org/whl/cu118
+
+# COPY --chown=user . /home/user
+
 ADD "https://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@api.github.com/repos/zqh0253/tmp_berf/commits?per_page=1" latest_commit
-RUN git clone https://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@github.com/zqh0253/tmp_berf.git berf
-COPY --chown=user . /home/user/berf
+RUN git clone https://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@github.com/zqh0253/tmp_berf.git /home/user/berf
 WORKDIR /home/user/berf
 
 RUN pip install --no-cache-dir --upgrade -r requirements.txt