ci: fix docker build by adding cuda compat ld load path to fix missing

libcuda.so.1
TabbyML · Jul 3, 2024 · a480375 · a480375
1 parent 26e0842
commit a480375
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/docker/Dockerfile.cuda b/docker/Dockerfile.cuda
@@ -27,6 +27,7 @@ RUN apt-get update && \
 # setup rust.
 RUN curl https://sh.rustup.rs -sSf | bash -s -- --default-toolchain ${RUST_TOOLCHAIN} -y
 ENV PATH="/root/.cargo/bin:${PATH}"
+ENV LD_LIBRARY_PATH="/usr/local/cuda/compat/:${LD_LIBRARY_PATH}"
 
 WORKDIR /root/workspace