Skip to content

Commit f2690e2

Browse files
authored
[docs] Expand on instructions for installing torch for CPU. (iree-org#19493)
Context: iree-org/iree-turbine#343 (comment) Screenshot: ![image](https://github.com/user-attachments/assets/7643cb54-4283-4de8-b8f5-c49d34efc2e9)
1 parent 8e86bcf commit f2690e2

File tree

1 file changed

+42
-23
lines changed

1 file changed

+42
-23
lines changed

docs/website/docs/guides/ml-frameworks/pytorch.md

Lines changed: 42 additions & 23 deletions
Original file line numberDiff line numberDiff line change
@@ -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

Comments
 (0)