zhangyiss/mlx
1
0
Fork 0
mirror of https://github.com/ml-explore/mlx.git synced 2025-08-03 01:06:43 +08:00
Code Issues Actions 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
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
3 changed files with 3 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
mlx/version.h
View File

@ -4,7 +4,7 @@
#define MLX_VERSION_MAJOR 0 #define MLX_VERSION_MAJOR 0
#define MLX_VERSION_MINOR 27 #define MLX_VERSION_MINOR 27
#define MLX_VERSION_PATCH 0 #define MLX_VERSION_PATCH 1
#define MLX_VERSION_NUMERIC \ #define MLX_VERSION_NUMERIC \
(100000 * MLX_VERSION_MAJOR + 1000 * MLX_VERSION_MINOR + MLX_VERSION_PATCH) (100000 * MLX_VERSION_MAJOR + 1000 * MLX_VERSION_MINOR + MLX_VERSION_PATCH)

1
python/scripts/repair_cuda.sh
View File

@ -5,6 +5,7 @@ auditwheel repair dist/* \
--exclude libcublas* \ --exclude libcublas* \
--exclude libnvrtc* \ --exclude libnvrtc* \
--exclude libcuda* \ --exclude libcuda* \
--exclude libcudnn* \
-w wheel_tmp -w wheel_tmp

2
setup.py
View File

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