revert default cuda install (#2465)

* revert default cuda install

* revert default cuda install
This commit is contained in:
Awni Hannun
2025-08-06 06:19:12 -07:00
committed by GitHub
parent 7bb96e4249
commit db5c7efcf6
3 changed files with 13 additions and 10 deletions

View File

@@ -75,8 +75,13 @@ macOS, run:
pip install mlx
```
On Linux, the same command (``pip install mlx``) will install the CUDA backend
by default. To install a CPU-only Linux package, run:
To install the CUDA backend on Linux, run:
```bash
pip install mlx[cuda]
```
To install a CPU-only Linux package, run:
```bash
pip install mlx[cpu]