File tree Expand file tree Collapse file tree 1 file changed +12
-0
lines changed Expand file tree Collapse file tree 1 file changed +12
-0
lines changed Original file line number Diff line number Diff line change @@ -49,6 +49,18 @@ RUN pip install --upgrade pip setuptools wheel build scikit-build-core[pyproject
4949# Copy source code to container
5050COPY . .
5151
52+ # Get LFS files (convert LFS pointers to actual files)
53+ RUN echo "📥 Fetching LFS files..." && \
54+ apt-get update && apt-get install -y --no-install-recommends git-lfs && \
55+ git lfs pull && \
56+ echo "✅ LFS files fetched successfully" && \
57+ ls -lh flash-attention/hopper/dist/*.whl && \
58+ echo "🧹 Cleaning up..." && \
59+ apt-get remove -y git-lfs && \
60+ apt-get autoremove -y && \
61+ rm -rf /var/lib/apt/lists/* && \
62+ echo "✅ Cleanup completed"
63+
5264# 🔧 设置 PyTorch 路径,让 CMake 能找到 Torch 配置
5365# 获取 PyTorch 安装路径并设置 CMAKE_PREFIX_PATH
5466RUN python -c "import torch; print(f'PyTorch installed at: {torch.__path__[0]}')" && \
You can’t perform that action at this time.
0 commit comments