diff --git a/docker/Dockerfile b/docker/Dockerfile
index 1b5d62fc7d77..b8db696a2a99 100644
--- a/docker/Dockerfile
+++ b/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"