|
@@ -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
|