diff --git a/pytorch/base.Dockerfile b/pytorch/base.Dockerfile
index 69fc6e6..6f08513 100644
--- a/pytorch/base.Dockerfile
+++ b/pytorch/base.Dockerfile
@@ -6,6 +6,7 @@ RUN set -eux; \
         python3-dev \
         python3-pip \
         python-is-python3 \
+        python3-mysqldb \
     ; \
     rm -rf /var/lib/apt/lists/*