diff --git a/centos-7/Dockerfile b/centos-7/Dockerfile
index a6dd91f31aa0664278511d1df448c7b312800f3c..161e85da6efc0d0547c6e155ac3bc952782f95ea 100644
--- a/centos-7/Dockerfile
+++ b/centos-7/Dockerfile
@@ -54,6 +54,8 @@ RUN yum install -y dkms && \
     cd /tmp && \
     rpm -ivh https://developer.download.nvidia.com/compute/cuda/repos/rhel7/x86_64/cuda-repo-rhel7-10.1.243-1.x86_64.rpm && \
     yum install -y cuda-10-1 && \
+    rpm -ivh http://developer.download.nvidia.com/compute/machine-learning/repos/rhel7/x86_64/nvidia-machine-learning-repo-rhel7-1.0.0-1.x86_64.rpm && \
+    yum install -y libcudnn7-devel-7.6.5.32-1.cuda10.1 && \
     yum clean all
 
 # FFmpeg