fix triton install?

NVIDIA · Dec 5, 2024 · f4ff8ce · f4ff8ce
1 parent d0ce65b
commit f4ff8ce
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/Dockerfile.arm b/Dockerfile.arm
@@ -47,9 +47,9 @@ RUN CAUSAL_CONV1D_FORCE_BUILD=TRUE pip --disable-pip-version-check --no-cache-di
 
 # Manually build triton (mamba dependency) - no existing ARM build
 RUN git clone https://github.com/triton-lang/triton.git && \
-    cd triton && \
     pip install ninja cmake wheel pybind11 && \
-    pip install python
+    cd triton/python && \
+    pip install .
 
 # Mamba dependancy installation
 RUN pip --disable-pip-version-check --no-cache-dir install \