From 039363b819646fab09fb73b12e787152de4c58ea Mon Sep 17 00:00:00 2001 From: Donato Capitella Date: Sat, 14 Mar 2026 13:41:09 +0000 Subject: [PATCH] feat: set LD_LIBRARY_PATH to include ROCm library directories. --- Dockerfile | 1 + 1 file changed, 1 insertion(+) diff --git a/Dockerfile b/Dockerfile index df1c427..9b7120d 100644 --- a/Dockerfile +++ b/Dockerfile @@ -34,6 +34,7 @@ WORKDIR /opt # Flash-Attention ENV FLASH_ATTENTION_TRITON_AMD_ENABLE="TRUE" +ENV LD_LIBRARY_PATH="/opt/rocm/lib:/opt/rocm/lib64:$LD_LIBRARY_PATH" RUN git clone https://github.com/ROCm/flash-attention.git &&\ cd flash-attention &&\