feat: set LD_LIBRARY_PATH to include ROCm library directories. · 039363b819 - amd-strix-halo-vllm-toolboxes - BadStorm.xyz - Code Hub

AI/amd-strix-halo-vllm-toolboxes

feat: set LD_LIBRARY_PATH to include ROCm library directories.

Przeglądaj źródła

This commit is contained in:

Donato Capitella

2026-03-14 13:41:09 +00:00

rodzic cf2fd6ec11

commit 039363b819

1 zmienionych plików z 1 dodań i 0 usunięć

									
										Dockerfile
									
		+1
		
												Wyświetl plik
												
					@@ -34,6 +34,7 @@ WORKDIR /opt

					# Flash-Attention

					# Flash-Attention

					ENV FLASH_ATTENTION_TRITON_AMD_ENABLE="TRUE"

					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 &&\ 

					RUN git clone https://github.com/ROCm/flash-attention.git &&\ 

					  cd flash-attention &&\

					  cd flash-attention &&\