From a1cc9ad1f0a3ea3e3a1c8163c5ff3c27065a9be4 Mon Sep 17 00:00:00 2001
From: Gerard Escalante <gerard@saygo.ca>
Date: Mon, 16 Nov 2020 17:11:10 +0000
Subject: [PATCH] Revert one other change

---
 docker/Dockerfile.amd64nvidia | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/docker/Dockerfile.amd64nvidia b/docker/Dockerfile.amd64nvidia
index 96fb62d06..8714f70ad 100644
--- a/docker/Dockerfile.amd64nvidia
+++ b/docker/Dockerfile.amd64nvidia
@@ -39,7 +39,7 @@ RUN echo "/usr/local/nvidia/lib" >> /etc/ld.so.conf.d/nvidia.conf && \
     echo "/usr/local/nvidia/lib64" >> /etc/ld.so.conf.d/nvidia.conf
 
 ENV PATH /usr/local/nvidia/bin:/usr/local/cuda/bin:${PATH}
-ENV LD_LIBRARY_PATH /usr/local/cuda/lib64
+ENV LD_LIBRARY_PATH /usr/local/nvidia/lib:/usr/local/nvidia/lib64
 
 # nvidia-container-runtime
 ENV NVIDIA_VISIBLE_DEVICES all