x
This commit is contained in:
+1
-1
@@ -28,7 +28,7 @@ ENV DEBIAN_FRONTEND=noninteractive \
|
|||||||
FLASH_ATTENTION_TRITON_AMD_ENABLE=TRUE \
|
FLASH_ATTENTION_TRITON_AMD_ENABLE=TRUE \
|
||||||
MINERU_MODEL_SOURCE=huggingface \
|
MINERU_MODEL_SOURCE=huggingface \
|
||||||
TORCH_ROCM_AOTRITON_ENABLE_EXPERIMENTAL=1 \
|
TORCH_ROCM_AOTRITON_ENABLE_EXPERIMENTAL=1 \
|
||||||
HSA_ENABLE_SDMA=1 \ # 启用系统 DMA 加速 GPU 显存传输
|
HSA_ENABLE_SDMA=1 \
|
||||||
VLLM_TARGET_DEVICE=rocm
|
VLLM_TARGET_DEVICE=rocm
|
||||||
|
|
||||||
WORKDIR /opt
|
WORKDIR /opt
|
||||||
|
|||||||
Reference in New Issue
Block a user