diff --git a/docker/Dockerfile b/docker/Dockerfile index ec6bfc5df..7ac13e771 100644 --- a/docker/Dockerfile +++ b/docker/Dockerfile @@ -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