fix(torch): Install pybind11 before attempting to build Triton

coreweave · Nov 26, 2024 · 762021f · 762021f
1 parent 626b44d
commit 762021f
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/torch/Dockerfile b/torch/Dockerfile
@@ -307,6 +307,7 @@ ARG BUILD_MAX_JOBS=""
 RUN --mount=type=bind,from=triton-downloader,source=/git/triton,target=triton/,rw \
     --mount=type=cache,target=/ccache \
     if [ "$BUILD_TRITON" = '1' ]; then \
+      pip3 install --no-cache-dir pybind11 && \
       export MAX_JOBS="${BUILD_MAX_JOBS:-$(./scale.sh "$(./effective_cpu_count.sh)" 3 32)}" && \
       cd triton/python && \
       python3 -m pip wheel -w ../../dist/ --no-build-isolation --no-deps -vv . && \