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:
parent
592feef04a
commit
1b753da112
|
|
@ -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 && \
|
||||
|
|
|
|||
Loading…
Reference in New Issue