use env

Dockerfile

@@ -2,6 +2,8 @@
 #FROM nvidia/cuda:11.8.0-devel-ubuntu22.04
 FROM ubuntu:latest
 
+ENV PIP_BREAK_SYSTEM_PACKAGES=1
+
 # Install dependencies
 RUN --mount=type=cache,target=/var/cache/apt,sharing=locked \
     --mount=type=cache,target=/var/lib/apt,sharing=locked \
@@ -26,7 +28,7 @@ RUN --mount=type=cache,target=/var/cache/apt,sharing=locked \
 #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 --break-system-packages --upgrade pip
+RUN pip install --upgrade pip
 RUN GIT_SSL_NO_VERIFY=true pip install --break-system-packages git+https://github.com/llvm/torch-mlir.git
 
 # Add PyTorch-MLIR tools to PATH