From 113d082cc4bb464b5cc6b0184289ea9afbdfd01b Mon Sep 17 00:00:00 2001
From: Michael Pilosov <40366263+mathematicalmichael@users.noreply.github.com>
Date: Mon, 31 Aug 2020 21:23:02 -0600
Subject: [PATCH] closes #25 by adding libcudnn7-dev

---
 src/Dockerfile.gpulibs | 3 +++
 1 file changed, 3 insertions(+)

diff --git a/src/Dockerfile.gpulibs b/src/Dockerfile.gpulibs
index 339f37c..ff7fb15 100644
--- a/src/Dockerfile.gpulibs
+++ b/src/Dockerfile.gpulibs
@@ -15,6 +15,9 @@ RUN pip install --upgrade pip && \
 RUN conda install --quiet --yes \
     pyyaml mkl mkl-include setuptools cmake cffi typing
 
+# Deep-NN libs
+RUN apt-get update && apt-get install -y libcudnn7-dev && rm -rf /var/lib/apt/lists/* 
+
 # Check compatibility here: https://pytorch.org/get-started/locally/
 RUN conda install --quiet --yes \
      pytorch \