mirror of https://github.com/llvm/torch-mlir
Update README.md and put Python 3.10 a prerequisite (#1821)
* Update README.md and put Python 3.10 a prerequisite * Update README.md * Follow up with comments * Update README.mdpull/1740/merge snapshot-20230126.730
parent
506aceae7f
commit
27a31157b9
21
README.md
21
README.md
|
@ -43,15 +43,26 @@ We have few paths to lower down to the Torch MLIR Dialect.
|
|||
|
||||
## Install torch-mlir snapshot
|
||||
|
||||
This installs a pre-built snapshot of torch-mlir for Python 3.7/3.8/3.9/3.10 on Linux and macOS.
|
||||
At the time of writing, we release pre-built snapshot of torch-mlir for Python 3.10 on Linux and macOS.
|
||||
|
||||
If you have Python 3.10, the following commands initialize a virtual environment.
|
||||
```shell
|
||||
python -m venv mlir_venv
|
||||
python3.10 -m venv mlir_venv
|
||||
source mlir_venv/bin/activate
|
||||
# Some older pip installs may not be able to handle the recent PyTorch deps
|
||||
```
|
||||
|
||||
Or, if you want to switch over multiple versions of Python using conda, you can create a conda environment with Python 3.10.
|
||||
```shell
|
||||
conda create -n torch-mlir python=3.10
|
||||
conda activate torch-mlir
|
||||
python -m pip install --upgrade pip
|
||||
pip install --pre torch-mlir torchvision -f https://llvm.github.io/torch-mlir/package-index/ --extra-index-url https://download.pytorch.org/whl/nightly/cpu
|
||||
# This will install the corresponding torch and torchvision nightlies
|
||||
```
|
||||
|
||||
Then, we can install torch-mlir with the corresponding torch and torchvision nightlies.
|
||||
```
|
||||
pip install --pre torch-mlir torchvision \
|
||||
-f https://llvm.github.io/torch-mlir/package-index/
|
||||
--extra-index-url https://download.pytorch.org/whl/nightly/cpu
|
||||
```
|
||||
|
||||
## Demos
|
||||
|
|
Loading…
Reference in New Issue