added cudnn environment variables

EPCCed · May 22, 2024 · a8bb42c · a8bb42c
1 parent 9ac54a4
commit a8bb42c
Showing 1 changed file with 3 additions and 0 deletions.
diff --git 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" ]