AI/amd-strix-halo-vllm-toolboxes
2
0
Garpu 0
Kode Masalah Tarik Permintaan Actions Packages Projects Rilis Wiki Kegiatan

feat: set LD_LIBRARY_PATH to include ROCm library directories.

Telusuri Sumber
This commit is contained in:
Donato Capitella
2026-03-14 13:41:09 +00:00
orang tua cf2fd6ec11
melakukan 039363b819
1 mengubah file dengan 1 tambahan dan 0 penghapusan
Download Patch File Download Diff File Expand all files Collapse all files
Dockerfile
+1
Melihat File
@@ -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 &&\