diff --git a/docker/Dockerfile b/docker/Dockerfile index cf6e5d4..a82145e 100644 --- a/docker/Dockerfile +++ b/docker/Dockerfile @@ -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