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" ]