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

fix cuda pypi package (#2423)

Browse Source 
* fix cuda pypi package

* patch bump
This commit is contained in:
Awni Hannun
2025-07-25 15:20:29 -07:00
committed by GitHub
parent d1165b215e
commit 4ad53414dd
3 changed files with 3 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
setup.py
View File

@@ -289,7 +289,7 @@ if __name__ == "__main__":
install_requires += [
"nvidia-cublas-cu12==12.9.*",
"nvidia-cuda-nvrtc-cu12==12.9.*",
"nvidia-cudnn-cu12==12.9.*",
"nvidia-cudnn-cu12==9.*",
]
else:
name = "mlx-cpu"