Merge pull request #436 from puneetmatharu/remove-docker-username-arg

Remove `USERNAME` arg from PyTorch Docker build

Author: puneetmatharu

ML-Frameworks/pytorch-aarch64/Dockerfile

@@ -12,12 +12,11 @@ ARG DOCKER_IMAGE_MIRROR=""
 # ============================
 FROM ${DOCKER_IMAGE_MIRROR}${BASE_IMAGE} AS workshop
 
-ARG USERNAME
 ARG TORCH_WHEEL
 ARG TORCH_AO_WHEEL
 
 ENV DEBIAN_FRONTEND=noninteractive
-ENV DOCKER_USER=${USERNAME}
+ENV DOCKER_USER=debian
 
 RUN test "$(arch)" = "aarch64"
 
@@ -71,9 +70,8 @@ COPY --chown=$DOCKER_USER:$DOCKER_USER pytorch/test /home/$DOCKER_USER/pytorch/t
 # ============================
 FROM ${DOCKER_IMAGE_MIRROR}${BASE_IMAGE}
 
-ARG USERNAME
 ENV DEBIAN_FRONTEND=noninteractive
-ENV DOCKER_USER=${USERNAME}
+ENV DOCKER_USER=debian
 
 # Runtime OS bits + UI
 RUN set -eux && \

ML-Frameworks/pytorch-aarch64/dockerize.sh

@@ -22,17 +22,13 @@ if ! [ -e "$1" ] || ! [ -e "$2" ]; then
     exit 1
 fi
 
-IMAGE_USERNAME="${USERNAME:-$(. /etc/os-release && echo "$ID")}"
-echo "USERNAME=$IMAGE_USERNAME"
-
 docker buildx \
     build --load \
     -t toolsolutions-pytorch:latest  \
     --build-context rootdir=../.. \
     --build-arg DOCKER_IMAGE_MIRROR \
     --build-arg TORCH_WHEEL=$1 \
     --build-arg TORCH_AO_WHEEL=$2 \
-    --build-arg USERNAME="$IMAGE_USERNAME" \
     .
 
 [[ $* == *--build-only* ]] && exit 0