Skip to content

Commit 10d74ba

Browse files
committed
Remove flash-attn to unblock build (can add back later)
1 parent f333930 commit 10d74ba

File tree

1 file changed

+1
-5
lines changed

1 file changed

+1
-5
lines changed

docker/dockerfiles/Dockerfile.onnx.jetson.6.2.0

Lines changed: 1 addition & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -123,8 +123,7 @@ ENV CUDA_HOME=/usr/local/cuda \
123123
LD_LIBRARY_PATH=/usr/local/cuda/lib64:$LD_LIBRARY_PATH
124124

125125
# Install Python dependencies (torch/torchvision/numpy already installed from source)
126-
# Use --no-deps to prevent uv from reinstalling torch/torchvision
127-
RUN uv pip install --system --break-system-packages --index-strategy unsafe-best-match --no-deps \
126+
RUN uv pip install --system --break-system-packages --index-strategy unsafe-best-match \
128127
-r _requirements.txt \
129128
-r requirements.http.txt \
130129
-r requirements.clip.txt \
@@ -141,9 +140,6 @@ RUN uv pip install --system --break-system-packages --index-strategy unsafe-best
141140
packaging \
142141
&& rm -rf ~/.cache/uv
143142

144-
# Install flash-attn separately with regular pip (requires CUDA-enabled torch)
145-
RUN python3 -m pip install --break-system-packages --no-build-isolation flash-attn==2.8.2
146-
147143
# Build onnxruntime with numpy 2.x
148144
WORKDIR /build/onnxruntime
149145
RUN git clone --recursive --branch v${ONNXRUNTIME_VERSION} https://github.com/microsoft/onnxruntime.git && \

0 commit comments

Comments
 (0)