From 7986864cb09f78e87ee11c2c7cddd50e8680d170 Mon Sep 17 00:00:00 2001
From: Moritz Maxeiner <mm@ucw.sh>
Date: Mon, 2 Mar 2020 14:26:21 +0100
Subject: [PATCH] centos-7: Add cuDNN
---
centos-7/Dockerfile | 2 ++
1 file changed, 2 insertions(+)
diff --git a/centos-7/Dockerfile b/centos-7/Dockerfile
index a6dd91f..161e85d 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
--
GitLab