Skip to content

GitLab

Unverified Commit 2a60ac91 authored by emricksini-h's avatar emricksini-h Committed by GitHub
Browse files

[Improvement] Persist CUDA compat libraries paths to prevent reset on `apt-get` (#30784) 


Signed-off-by: default avataremricksini-h <emrick.birivoutin@hcompany.ai>
parent 9e65bb4e
Hide whitespace changes
Inline Side-by-side
Showing with 4 additions and 4 deletions
docker/Dockerfile
View file @ 2a60ac91
......@@ -117,8 +117,8 @@ ENV UV_LINK_MODE=copy
# Verify GCC version
RUN gcc --version
# Workaround for triton/pytorch issues
RUN ldconfig /usr/local/cuda-$(echo $CUDA_VERSION | cut -d. -f1,2)/compat/
# Ensure CUDA compatibility library is loaded
RUN echo "/usr/local/cuda-$(echo "$CUDA_VERSION" | cut -d. -f1,2)/compat/" > /etc/ld.so.conf.d/00-cuda-compat.conf && ldconfig
# ============================================================
# SLOW-CHANGING DEPENDENCIES BELOW
......@@ -453,8 +453,8 @@ ENV UV_HTTP_TIMEOUT=500
ENV UV_INDEX_STRATEGY="unsafe-best-match"
ENV UV_LINK_MODE=copy
# Workaround for triton/pytorch issues
RUN ldconfig /usr/local/cuda-$(echo $CUDA_VERSION | cut -d. -f1,2)/compat/
# Ensure CUDA compatibility library is loaded
RUN echo "/usr/local/cuda-$(echo "$CUDA_VERSION" | cut -d. -f1,2)/compat/" > /etc/ld.so.conf.d/00-cuda-compat.conf && ldconfig
# ============================================================
# SLOW-CHANGING DEPENDENCIES BELOW
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment