Skip to content

Commit 7c4c088

Browse files
committed
add lfs install
1 parent f56c434 commit 7c4c088

File tree

1 file changed

+12
-0
lines changed

1 file changed

+12
-0
lines changed

Dockerfile

Lines changed: 12 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -49,6 +49,18 @@ RUN pip install --upgrade pip setuptools wheel build scikit-build-core[pyproject
4949
# Copy source code to container
5050
COPY . .
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
5466
RUN python -c "import torch; print(f'PyTorch installed at: {torch.__path__[0]}')" && \

0 commit comments

Comments
 (0)