docker : fix the publishing of the CUDA Docker image (#1704)

2023-12-30 21:12:31 +00:00 · 2023-12-30 21:12:31 +00:00 · f9ca90256b
parent 2623640cd6
commit f9ca90256b
1 changed files with 4 additions and 0 deletions
--- a/.devops/main-cuda.Dockerfile
+++ b/.devops/main-cuda.Dockerfile
@ -20,6 +20,10 @@ RUN apt-get update && \
    apt-get install -y build-essential \
    && rm -rf /var/lib/apt/lists/* /var/cache/apt/archives/*

+# Ref: https://stackoverflow.com/a/53464012
+ENV CUDA_MAIN_VERSION=12.3
+ENV LD_LIBRARY_PATH /usr/local/cuda-${CUDA_MAIN_VERSION}/compat:$LD_LIBRARY_PATH
+
 COPY .. .
 RUN make