x
This commit is contained in:
+1
-1
@@ -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
|
||||
|
||||
Reference in New Issue
Block a user