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

install linux with mlx[cuda] and mlx[cpu]

Browse Source
This commit is contained in:
Awni Hannun
2025-07-10 17:18:14 -07:00
parent 8347575ba1
commit 9381163788
3 changed files with 76 additions and 31 deletions
Download Patch File Download Diff File Expand all files Collapse all files

10
docs/src/install.rst
View File

@@ -38,8 +38,16 @@ and SM 7.0 (Volta) and up. To install MLX with CUDA support, run:
.. code-block:: shell
pip install mlx-cuda
pip install "mlx[cuda]"
CPU only (Linux)
^^^^^^^^^^^^^^^^
For a CPU-only version of MLX that runs on Linux use:
.. code-block:: shell
pip install "mlx[cpu]"
Troubleshooting
^^^^^^^^^^^^^^^