zhangyiss/mlx
1
0
Fork 0
mirror of https://github.com/ml-explore/mlx.git synced 2025-09-01 12:49:44 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

default install cuda on linux (#2462)

Browse Source
This commit is contained in:
Awni Hannun
2025-08-04 15:33:05 -07:00
committed by GitHub
parent 737dd6d1ac
commit 6ad0889c8a
3 changed files with 12 additions and 15 deletions
Download Patch File Download Diff File Expand all files Collapse all files

11
README.md
View File

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