zhangyiss/mlx
1
0
Fork 0
mirror of https://github.com/ml-explore/mlx.git synced 2025-09-09 22:16:43 +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

4
docs/src/install.rst
View File

@@ -30,7 +30,7 @@ MLX has a CUDA backend which you can install with:
.. code-block:: shell
pip install "mlx[cuda]"
pip install mlx
To install the CUDA package from PyPi your system must meet the following
requirements:
@@ -49,7 +49,7 @@ For a CPU-only version of MLX that runs on Linux use:
.. code-block:: shell
pip install "mlx[cpu]"
pip install mlx[cpu]
To install the CPU-only package from PyPi your system must meet the following
requirements: