diff --git a/README.md b/README.md
index fd8d848..2af2fc6 100644
--- a/README.md
+++ b/README.md
@@ -21,6 +21,7 @@ apt-get install python3-numpy python3-dev python3-pip python3-mock
 # If using a virtual environment, omit the --user argument
 pip3 install -U --user keras_applications==1.0.8 --no-deps
 pip3 install -U --user keras_preprocessing==1.1.0 --no-deps
+pip3 install portpicker
 ```
 
 ## TensorFlow on Raspberry Pi
diff --git a/build_tensorflow/Dockerfile b/build_tensorflow/Dockerfile
index 0ebf21a..68629d7 100644
--- a/build_tensorflow/Dockerfile
+++ b/build_tensorflow/Dockerfile
@@ -11,6 +11,7 @@ RUN dpkg --add-architecture armhf && dpkg --add-architecture arm64 \
 
 RUN pip3 install -U --user keras_applications==1.0.8 --no-deps \
     && pip3 install -U --user keras_preprocessing==1.1.0 --no-deps \
+    && pip3 install portpicker \
     && ldconfig
 
 RUN /bin/bash -c "update-alternatives --install /usr/bin/python python /usr/bin/python3 150"
diff --git a/build_tensorflow/Dockerfile.stretch b/build_tensorflow/Dockerfile.stretch
index c802665..b9eb4d2 100644
--- a/build_tensorflow/Dockerfile.stretch
+++ b/build_tensorflow/Dockerfile.stretch
@@ -17,6 +17,7 @@ RUN echo "deb http://ftp.us.debian.org/debian/ buster main contrib non-free" > /
 
 RUN pip3 install -U --user keras_applications==1.0.8 --no-deps \
     && pip3 install -U --user keras_preprocessing==1.1.0 --no-deps \
+    && pip3 install portpicker \
     && ldconfig
 
 RUN /bin/bash -c "update-alternatives --install /usr/bin/python python /usr/bin/python3 150"