[aarch64] Dockerfile support

Tested with base_image=ubuntu:focal
docker build -t --build-arg base_image=ubuntu:focal torch-xla -f docker/Dockerfile .

Author: snadampal

docker/Dockerfile

@@ -14,9 +14,12 @@ ARG bazel_jobs=""
 ARG git_clone="true"
 
 RUN apt-get update
-RUN apt-get install -y git sudo python-pip python3-pip
+RUN apt-get install -y git sudo python3-pip
 RUN git clone https://github.com/pytorch/pytorch
 
+RUN echo 'debconf debconf/frontend select Noninteractive' | debconf-set-selections
+RUN sudo apt-get install -y -q
+
 # Disable CUDA for PyTorch
 ENV USE_CUDA "0"