This commit is contained in:
2026-06-04 15:37:38 +08:00
parent 217985d85a
commit 030693cd68
+1
View File
@@ -188,6 +188,7 @@ RUN set -ex && \
-DHIP_ROOT_DIR=/opt/rocm \
-DROCM_PATH=/opt/rocm \
-DCMAKE_HIP_ARCHITECTURES=${ARCH} \
-DCMAKE_HIP_COMPILER=/opt/rocm/llvm/bin/clang++ \
-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 && \