diff --git a/container/Dockerfile.cpu b/container/Dockerfile.cpu
index 3b66597fe..ed1c15b72 100644
--- a/container/Dockerfile.cpu
+++ b/container/Dockerfile.cpu
@@ -59,12 +59,13 @@ RUN apt-get update && \
     openjdk-8-jdk-headless \
     python3-dev \
     gcc \
+    libgtk2.0-dev \
     && rm -rf /var/lib/apt/lists/*
 
 COPY --from=builder /workspace/python/dist/*.whl /home/model-server/
 
 RUN pip3 install --pre --no-cache-dir multi-model-server \
-    && pip3 install --no-cache-dir --prefer-binary numpy scipy xlrd Pillow boto3 six requests mxnet tensorflow \
+    && pip3 install --no-cache-dir --prefer-binary numpy scipy xlrd Pillow boto3 six requests mxnet tensorflow opencv-python \
     && pip3 install /home/model-server/dlr-*.whl \
     && rm -rf /root/.cache/pip
 
diff --git a/container/Dockerfile.gpu b/container/Dockerfile.gpu
index 1182a2da4..af2c522bd 100644
--- a/container/Dockerfile.gpu
+++ b/container/Dockerfile.gpu
@@ -66,12 +66,13 @@ RUN apt-get update && \
     openjdk-8-jdk-headless \
     python3-dev \
     gcc \
+    libgtk2.0-dev \
     && rm -rf /var/lib/apt/lists/*
 
 COPY --from=builder /workspace/python/dist/*.whl /home/model-server/
 
 RUN pip3 install --pre --no-cache-dir multi-model-server \
-    && pip3 install --no-cache-dir --prefer-binary numpy scipy xlrd Pillow boto3 six requests mxnet-cu100 tensorflow_gpu \
+    && pip3 install --no-cache-dir --prefer-binary numpy scipy xlrd Pillow boto3 six requests mxnet-cu100 tensorflow_gpu opencv-python \
     && pip3 install /home/model-server/dlr-*.whl \
     && rm -rf /root/.cache/pip