From cc96e89a1839c6ce535b2864dd275bf050268efd Mon Sep 17 00:00:00 2001 From: Rolf Morel Date: Thu, 14 Aug 2025 09:22:54 -0700 Subject: [PATCH] [ingress][torch-mlir] Fix install issue by using py3.12 in venv Uses `uv` packaga manager++ for managing venv. --- ingress/Torch-MLIR/install-virtualenv.sh | 15 +++++++++++---- 1 file changed, 11 insertions(+), 4 deletions(-) diff --git a/ingress/Torch-MLIR/install-virtualenv.sh b/ingress/Torch-MLIR/install-virtualenv.sh index 1ee362b..f48019c 100755 --- a/ingress/Torch-MLIR/install-virtualenv.sh +++ b/ingress/Torch-MLIR/install-virtualenv.sh @@ -11,10 +11,17 @@ fi # Install torch-mlir inside a virtual environment +echo "First ensure uv is installed" + +python -m pip install uv --upgrade + echo "Preparing the virtual environment" -python3 -m venv torch-mlir-venv +python -m uv venv torch-mlir-venv --python 3.12 + +#echo "Preparing the virtual environment" +#python3 -m venv torch-mlir-venv source torch-mlir-venv/bin/activate -python -m pip install --upgrade pip wheel +uv pip install --upgrade pip wheel # GPU support ("AMD", "NVIDIA", "Intel") EXTRA_INDEX_URL="" @@ -34,7 +41,7 @@ else fi echo "Installing torch and models" -pip3 install --pre torch torchvision torchrec transformers $EXTRA_INDEX_URL +uv pip install --pre torch torchvision torchrec transformers $EXTRA_INDEX_URL if [ $? != 0 ]; then exit 1 fi @@ -42,7 +49,7 @@ fi echo "Installing torch-mlir" # This only seems to work on Ubuntu -pip3 install --pre torch-mlir \ +uv pip install --pre torch-mlir \ --extra-index-url https://download.pytorch.org/whl/nightly/cpu \ -f https://github.com/llvm/torch-mlir-release/releases/expanded_assets/dev-wheels if [ $? != 0 ]; then