File tree Expand file tree Collapse file tree 1 file changed +1
-5
lines changed Expand file tree Collapse file tree 1 file changed +1
-5
lines changed Original file line number Diff line number Diff 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
148144WORKDIR /build/onnxruntime
149145RUN git clone --recursive --branch v${ONNXRUNTIME_VERSION} https://github.com/microsoft/onnxruntime.git && \
You can’t perform that action at this time.
0 commit comments