From a8bb42c34d17fa93fc7651508705e9038e0ca3f1 Mon Sep 17 00:00:00 2001
From: Dave McKay <d.mckay@epcc.ed.ac.uk>
Date: Wed, 22 May 2024 10:16:55 +0100
Subject: [PATCH] added cudnn environment variables

---
 files/container/gpu/Dockerfile | 3 +++
 1 file changed, 3 insertions(+)

diff --git a/files/container/gpu/Dockerfile b/files/container/gpu/Dockerfile
index 9246d2d..14e5489 100644
--- a/files/container/gpu/Dockerfile
+++ b/files/container/gpu/Dockerfile
@@ -112,6 +112,9 @@ COPY files/container/entrypoint.sh /entrypoint.sh
 
 # USER simlint
 
+ENV CUDNN_LIBRARY_PATH=/usr/lib/x86_64-linux-gnu
+ENV CUDNN_INCLUDE_PATH=/usr/include
+
 WORKDIR /
 
 ENTRYPOINT [ "/entrypoint.sh" ]