Fix ninja install

pytorch-labs · Dec 20, 2024 · 1e1b173 · 1e1b173
1 parent 5744fcd
commit 1e1b173
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/docker/tritonbench-nightly.dockerfile b/docker/tritonbench-nightly.dockerfile
@@ -50,7 +50,7 @@ RUN cd /workspace/tritonbench && \
 RUN sudo apt update && sudo apt-get install -y libnvidia-compute-550 patchelf patch
 
 # Workaround: installing Ninja from setup.py hits "Failed to decode METADATA with UTF-8" error
-RUN pip install ninja
+RUN . ${SETUP_SCRIPT} && pip install ninja
 
 # Install Tritonbench
 RUN cd /workspace/tritonbench && \