pip

Dockerfile

@@ -21,10 +21,10 @@ RUN --mount=type=cache,target=/var/cache/apt,sharing=locked \
 
 # Install pytorch-mlir
 
-RUN pip install --break-system-packages --upgrade pip setuptools
-RUN pip install --break-system-packages --pre torch-mlir torch \
+#RUN pip install --break-system-packages --pre torch-mlir torch \
   --extra-index-url https://download.pytorch.org/whl/nightly/cpu \
   -f https://github.com/llvm/torch-mlir-release/releases/expanded_assets/dev-wheels
+RUN pip install -e git+https://github.com/llvm/torch-mlir
 
 # Add PyTorch-MLIR tools to PATH
 ENV PATH="/root/.local/bin:${PATH}"