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

revert default cuda install (#2465)

Browse Source 
* 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
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
3 changed files with 13 additions and 10 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
README.md
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]

2
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
pip install mlx[cuda]
To install the CUDA package from PyPi your system must meet the following
requirements:

12
setup.py
View File

@ -274,13 +274,11 @@ if __name__ == "__main__":
# - Package name is back-end specific, e.g mlx-metal
if build_stage != 2:
if build_stage == 1:
install_requires += [
f'mlx-metal=={version}; platform_system == "Darwin"',
f'mlx-cuda=={version}; extra != "cpu" and platform_system == "Linux"',
]
extras["cpu"] = [
f'mlx-cpu=={version}; extra == "cpu" and platform_system == "Linux"'
]
install_requires.append(
f'mlx-metal=={version}; platform_system == "Darwin"'
)
extras["cuda"] = [f'mlx-cuda=={version}; platform_system == "Linux"']
extras["cpu"] = [f'mlx-cpu=={version}; platform_system == "Linux"']
_setup(
name="mlx",