This commit is contained in:
2026-06-03 16:50:52 +08:00
parent 3550efcd48
commit 7d4d22aaa5
+1 -1
View File
@@ -28,7 +28,7 @@ ENV DEBIAN_FRONTEND=noninteractive \
FLASH_ATTENTION_TRITON_AMD_ENABLE=TRUE \
MINERU_MODEL_SOURCE=huggingface \
TORCH_ROCM_AOTRITON_ENABLE_EXPERIMENTAL=1 \
HSA_ENABLE_SDMA=1 \ # 启用系统 DMA 加速 GPU 显存传输
HSA_ENABLE_SDMA=1 \
VLLM_TARGET_DEVICE=rocm
WORKDIR /opt