@@ -62,37 +62,56 @@ graph LR
6262
6363## :octicons-download-16: Prerequisites
6464
65- We recommend first installing a recent version of PyTorch for CPU by following
66- the [ official instructions] ( https://pytorch.org/get-started/locally/ ) .
65+ 1 . First install a recent version of PyTorch by following
66+ the [ official instructions] ( https://pytorch.org/get-started/locally/ ) :
6767
68- ``` shell
69- python -m pip install \
70- --index-url https://download.pytorch.org/whl/test/cpu torch> =2.3.0
71- ```
68+ === ":fontawesome-brands-linux: Linux"
7269
73- Install iree-turbine:
70+ ``` shell
71+ python -m pip install torch --index-url https://download.pytorch.org/whl/test/cpu
72+ ```
7473
75- === ":octicons-package-16: Stable releases "
74+ === ":fontawesome-brands-apple: macOS "
7675
77- Stable release packages are
78- [published to PyPI](https://pypi.org/project/iree-turbine/).
76+ ``` shell
77+ python -m pip install torch
78+ ```
7979
80- ``` shell
81- python -m pip install iree-turbine
82- ```
80+ === ":fontawesome-brands-windows: Windows"
8381
84- === ":octicons-beaker-16: Nightly pre-releases"
82+ ``` shell
83+ python -m pip install torch
84+ ```
8585
86- Nightly pre-releases are published on
87- [GitHub releases](https://github.com/iree-org/iree-turbine/releases/tag/dev-wheels).
86+ !!! tip
8887
89- ``` shell hl_lines="2-4"
90- python -m pip install \
91- --find-links https://iree.dev/pip-release-links.html \
92- --pre \
93- --upgrade \
94- iree-turbine
95- ```
88+ IREE includes its own GPU support, so we recommend the CPU versions of
89+ PyTorch. You can install CUDA or ROCm as you wish, but those packages
90+ can be quite large.
91+
92+ 2 . Then install iree-turbine:
93+
94+ === ":octicons-package-16: Stable releases"
95+
96+ Stable release packages are
97+ [published to PyPI](https://pypi.org/project/iree-turbine/).
98+
99+ ``` shell
100+ python -m pip install iree-turbine
101+ ```
102+
103+ === ":octicons-beaker-16: Nightly pre-releases"
104+
105+ Nightly pre-releases are published on
106+ [GitHub releases](https://github.com/iree-org/iree-turbine/releases/tag/dev-wheels).
107+
108+ ``` shell hl_lines="2-4"
109+ python -m pip install \
110+ --find-links https://iree.dev/pip-release-links.html \
111+ --pre \
112+ --upgrade \
113+ iree-turbine
114+ ```
96115
97116## :octicons-flame-16: Just-in-time (JIT) execution
98117
0 commit comments