Add a ld.so.conf entry to find CUDA.

Author: maleadt

runner/Dockerfile.arch

@@ -12,6 +12,9 @@ RUN pacman -Suy --noconfirm --needed \
     # clean-up
     && find /var/cache/pacman/ -type f -delete
 
+# to find CUDA
+RUN echo "/usr/lib/x86_64-linux-gnu" > /etc/ld.so.conf.d/cuda.conf && ldconfig
+
 RUN mkdir /storage /cache
 
 COPY ./entrypoint.sh /

runner/Dockerfile.ubuntu

@@ -14,6 +14,9 @@ RUN apt-get update && apt-get install --no-install-recommends -y \
     # clean-up
     && rm -rf /var/lib/apt/lists/*
 
+# to find CUDA
+RUN echo "/usr/lib/x86_64-linux-gnu" > /etc/ld.so.conf.d/cuda.conf && ldconfig
+
 RUN mkdir /storage /cache
 
 COPY ./entrypoint.sh /