File tree Expand file tree Collapse file tree 1 file changed +11
-4
lines changed Expand file tree Collapse file tree 1 file changed +11
-4
lines changed Original file line number Diff line number Diff line change 1111
1212
1313# Install torch-mlir inside a virtual environment
14+ echo " First ensure uv is installed"
15+
16+ python -m pip install uv --upgrade
17+
1418echo " Preparing the virtual environment"
15- python3 -m venv torch-mlir-venv
19+ python -m uv venv torch-mlir-venv --python 3.12
20+
21+ # echo "Preparing the virtual environment"
22+ # python3 -m venv torch-mlir-venv
1623source torch-mlir-venv/bin/activate
17- python -m pip install --upgrade pip wheel
24+ uv pip install --upgrade pip wheel
1825
1926# GPU support ("AMD", "NVIDIA", "Intel")
2027EXTRA_INDEX_URL=" "
3441fi
3542
3643echo " Installing torch and models"
37- pip3 install --pre torch torchvision torchrec transformers $EXTRA_INDEX_URL
44+ uv pip install --pre torch torchvision torchrec transformers $EXTRA_INDEX_URL
3845if [ $? != 0 ]; then
3946 exit 1
4047fi
4148
4249
4350echo " Installing torch-mlir"
4451# This only seems to work on Ubuntu
45- pip3 install --pre torch-mlir \
52+ uv pip install --pre torch-mlir \
4653 --extra-index-url https://download.pytorch.org/whl/nightly/cpu \
4754 -f https://github.com/llvm/torch-mlir-release/releases/expanded_assets/dev-wheels
4855if [ $? != 0 ]; then
You can’t perform that action at this time.
0 commit comments