diff --git a/Dockerfile b/Dockerfile
index df762f58..97a68077 100644
--- a/Dockerfile
+++ b/Dockerfile
@@ -22,6 +22,8 @@ RUN sed -i "s/httpredir.debian.org/debian.uchicago.edu/" /etc/apt/sources.list &
     # https://lightgbm.readthedocs.io/en/latest/GPU-Tutorial.html#build-lightgbm
     apt-get install -y build-essential unzip cmake && \
     apt-get install -y libboost-dev libboost-program-options-dev libboost-system-dev libboost-thread-dev libboost-math-dev libboost-test-dev libboost-python-dev libboost-filesystem-dev zlib1g-dev && \
+    # b/182601974: ssh client was removed from the base image but is required for packages such as stable-baselines.
+    apt-get install -y openssh-client && \
     pip install --upgrade pip && \
     /tmp/clean-layer.sh