docker : libcuda.so.1 in PATH (#1966)

ggerganov · Mar 20, 2024 · 79d5765 · 79d5765
1 parent 04e4809
commit 79d5765
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/.devops/main-cuda.Dockerfile b/.devops/main-cuda.Dockerfile
@@ -28,6 +28,8 @@ COPY .. .
 RUN make
 
 FROM ${BASE_CUDA_RUN_CONTAINER} AS runtime
+ENV CUDA_MAIN_VERSION=12.3
+ENV LD_LIBRARY_PATH /usr/local/cuda-${CUDA_MAIN_VERSION}/compat:$LD_LIBRARY_PATH
 WORKDIR /app
 
 RUN apt-get update && \