fix: stop shadowing proper cuda libraries in runtime docker build

The builder NEEDS the library for building because it has no GPU, but forcing the same in the RUNTIME build BREAKS GPU support.
This commit is contained in:
Fred N. Garvin 2026-08-13 13:15:25 -05:00 committed by GitHub
parent 592feef04a
commit 1b753da112
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 2 additions and 2 deletions

View File

@ -19,6 +19,8 @@ RUN apt-get update && \
&& apt-get clean \
&& rm -rf /var/lib/apt/lists/* /var/cache/apt/archives/*
# The build stage has no real GPU/driver, so the linker needs the
# forward-compat libcuda stub to resolve CUDA driver-API symbols.
# Ref: https://stackoverflow.com/a/53464012
ENV CUDA_MAIN_VERSION=13.0
ENV LD_LIBRARY_PATH /usr/local/cuda-${CUDA_MAIN_VERSION}/compat:$LD_LIBRARY_PATH
@ -34,8 +36,6 @@ RUN find /app/build -name "*.o" -delete && \
rm -rf /app/build/_deps
FROM ${BASE_CUDA_RUN_CONTAINER} AS runtime
ENV CUDA_MAIN_VERSION=13.0
ENV LD_LIBRARY_PATH /usr/local/cuda-${CUDA_MAIN_VERSION}/compat:$LD_LIBRARY_PATH
WORKDIR /app
RUN apt-get update && \