x
This commit is contained in:
+7
-2
@@ -134,9 +134,11 @@ RUN apt-get update && apt-get install -y --no-install-recommends \
|
||||
# 阶段 7:amd-aiter + flash_attn
|
||||
# ===========================================================================
|
||||
RUN set -ex && \
|
||||
# git 通过代理访问 GitHub(国内不通,其他源已用国内镜像)
|
||||
# git 通过代理访问 GitHub
|
||||
git config --global http.proxy ${GIT_PROXY} && \
|
||||
git config --global https.proxy ${GIT_PROXY} && \
|
||||
git config --global http.version HTTP/1.1 && \
|
||||
git config --global http.postBuffer 524288000 && \
|
||||
# aiter(AMD 优化的 attention 算子)
|
||||
cd /opt && git clone --recursive --depth 1 https://github.com/ROCm/aiter.git && \
|
||||
${VENV}/bin/pip install --no-cache-dir -e /opt/aiter && \
|
||||
@@ -151,9 +153,12 @@ RUN set -ex && \
|
||||
# 阶段 8a:编译 vllm(git clone + cmake + ninja,最耗时,单独一层缓存)
|
||||
# ===========================================================================
|
||||
RUN set -ex && \
|
||||
# git 通过代理访问 GitHub
|
||||
# git 通过代理访问 GitHub(国内不通)
|
||||
git config --global http.proxy ${GIT_PROXY} && \
|
||||
git config --global https.proxy ${GIT_PROXY} && \
|
||||
# 代理下大仓库容易断,改用 HTTP/1.1 + 加大 buffer
|
||||
git config --global http.version HTTP/1.1 && \
|
||||
git config --global http.postBuffer 524288000 && \
|
||||
# setuptools 升级(PEP 639 兼容)
|
||||
${VENV}/bin/pip install --no-cache-dir -U \
|
||||
"setuptools>=77.0.3" setuptools_scm setuptools_rust wheel && \
|
||||
|
||||
Reference in New Issue
Block a user