Revert one other change

blakeblackshear · Nov 17, 2020 · a1cc9ad · a1cc9ad
1 parent 29e8aa4
commit a1cc9ad
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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