From f9ca90256bf691642407e589db1a36562c461db7 Mon Sep 17 00:00:00 2001 From: bobqianic <129547291+bobqianic@users.noreply.github.com> Date: Sat, 30 Dec 2023 21:12:31 +0000 Subject: [PATCH] docker : fix the publishing of the CUDA Docker image (#1704) --- .devops/main-cuda.Dockerfile | 4 ++++ 1 file changed, 4 insertions(+) diff --git a/.devops/main-cuda.Dockerfile b/.devops/main-cuda.Dockerfile index 0ce7642..9345241 100644 --- 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