cjw/mineru-rocm
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

x

Browse Source
This commit is contained in:
chenjw28
2026-06-04 14:17:37 +08:00
parent 69b583340b
commit 4553b6c073
1 changed files with 7 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files
docker/Dockerfile
+7 -2
View File
@@ -134,9 +134,11 @@ RUN apt-get update && apt-get install -y --no-install-recommends \
# 阶段 7：amd-aiter + flash_attn # 阶段 7：amd-aiter + flash_attn
# =========================================================================== # ===========================================================================
RUN set -ex && \ RUN set -ex && \
# git 通过代理访问 GitHub（国内不通，其他源已用国内镜像） # git 通过代理访问 GitHub
git config --global http.proxy ${GIT_PROXY} && \ git config --global http.proxy ${GIT_PROXY} && \
git config --global https.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 算子） # aiter（AMD 优化的 attention 算子）
cd /opt && git clone --recursive --depth 1 https://github.com/ROCm/aiter.git && \ cd /opt && git clone --recursive --depth 1 https://github.com/ROCm/aiter.git && \
${VENV}/bin/pip install --no-cache-dir -e /opt/aiter && \ ${VENV}/bin/pip install --no-cache-dir -e /opt/aiter && \
@@ -151,9 +153,12 @@ RUN set -ex && \
# 阶段 8a：编译 vllm（git clone + cmake + ninja，最耗时，单独一层缓存） # 阶段 8a：编译 vllm（git clone + cmake + ninja，最耗时，单独一层缓存）
# =========================================================================== # ===========================================================================
RUN set -ex && \ RUN set -ex && \
# git 通过代理访问 GitHub # git 通过代理访问 GitHub（国内不通）
git config --global http.proxy ${GIT_PROXY} && \ git config --global http.proxy ${GIT_PROXY} && \
git config --global https.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 兼容） # setuptools 升级（PEP 639 兼容）
${VENV}/bin/pip install --no-cache-dir -U \ ${VENV}/bin/pip install --no-cache-dir -U \
"setuptools>=77.0.3" setuptools_scm setuptools_rust wheel && \ "setuptools>=77.0.3" setuptools_scm setuptools_rust wheel && \