diff --git a/docker/Dockerfile b/docker/Dockerfile index d573e74..513480a 100644 --- a/docker/Dockerfile +++ b/docker/Dockerfile @@ -192,14 +192,6 @@ RUN set -ex && \ printf 'include(/opt/rocm/lib/cmake/rocblas/rocblas-config.cmake)\nif(TARGET roc::rocblas AND NOT TARGET hip::hipblas)\n add_library(hip::hipblas ALIAS roc::rocblas)\nendif()\n' \ > /opt/rocm/lib/cmake/hipblas/hipblas-config.cmake && \ # cmake 配置 - # ---- 诊断 HIP cmake(构建成功后删除)---- - echo "=== HIP cmake files ===" && \ - (find /opt/rocm/lib/cmake -iname "*hip*" 2>/dev/null || true) && \ - echo "=== All cmake packages ===" && \ - (ls /opt/rocm/lib/cmake/ || true) && \ - echo "=== hipcc check ===" && \ - (which hipcc && hipcc --version || true) && \ - echo "=== HIP diagnostic end ===" && \ mkdir -p /opt/vllm_build && \ cmake -S /opt/vllm -B /opt/vllm_build -G Ninja \ -DCMAKE_BUILD_TYPE=RelWithDebInfo \ @@ -209,6 +201,8 @@ RUN set -ex && \ -DROCM_PATH=/opt/rocm \ -DCMAKE_HIP_ARCHITECTURES=${ARCH} \ -DCMAKE_HIP_COMPILER=/opt/rocm/llvm/bin/clang++ \ + -DHIP_COMPILER=/opt/rocm/llvm/bin/clang++ \ + -DHIP_PATH=/opt/rocm \ -DCMAKE_PREFIX_PATH="/opt/rocm;${VENV}/lib/python${PYTHON_VER}/site-packages/torch/share/cmake" && \ # ninja 编译(-j8,32GB 内存;若 < 16GB 请改为 -j2) cd /opt/vllm_build && ninja -j8 && \