[Improvement] Persist CUDA compat libraries paths to prevent reset on apt-get (#30784)
Signed-off-by: emricksini-h <emrick.birivoutin@hcompany.ai>
This commit is contained in:
@@ -117,8 +117,8 @@ ENV UV_LINK_MODE=copy
|
|||||||
# Verify GCC version
|
# Verify GCC version
|
||||||
RUN gcc --version
|
RUN gcc --version
|
||||||
|
|
||||||
# Workaround for triton/pytorch issues
|
# Ensure CUDA compatibility library is loaded
|
||||||
RUN ldconfig /usr/local/cuda-$(echo $CUDA_VERSION | cut -d. -f1,2)/compat/
|
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
|
# SLOW-CHANGING DEPENDENCIES BELOW
|
||||||
@@ -453,8 +453,8 @@ ENV UV_HTTP_TIMEOUT=500
|
|||||||
ENV UV_INDEX_STRATEGY="unsafe-best-match"
|
ENV UV_INDEX_STRATEGY="unsafe-best-match"
|
||||||
ENV UV_LINK_MODE=copy
|
ENV UV_LINK_MODE=copy
|
||||||
|
|
||||||
# Workaround for triton/pytorch issues
|
# Ensure CUDA compatibility library is loaded
|
||||||
RUN ldconfig /usr/local/cuda-$(echo $CUDA_VERSION | cut -d. -f1,2)/compat/
|
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
|
# SLOW-CHANGING DEPENDENCIES BELOW
|
||||||
|
|||||||
Reference in New Issue
Block a user