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.

此提交包含在：

Donato Capitella

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

父節點 cf2fd6ec11

當前提交 039363b819

共有 1 個檔案被更改，包括 1 行新增 和 0 行删除

									
										Dockerfile
									
		+1
		
												查看文件
												
					@@ -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 &&\