fix cuda pypi package (#2423)

awni · web-flow · commit 4ad53414dd6e · 2025-07-25T15:20:29.000-07:00
* fix cuda pypi package

* patch bump
diff --git a/mlx/version.h b/mlx/version.h
@@ -4,7 +4,7 @@
 
 #define MLX_VERSION_MAJOR 0
 #define MLX_VERSION_MINOR 27
-#define MLX_VERSION_PATCH 0
+#define MLX_VERSION_PATCH 1
 #define MLX_VERSION_NUMERIC \
   (100000 * MLX_VERSION_MAJOR + 1000 * MLX_VERSION_MINOR + MLX_VERSION_PATCH)
 
diff --git a/python/scripts/repair_cuda.sh b/python/scripts/repair_cuda.sh
@@ -5,6 +5,7 @@ auditwheel repair dist/* \
   --exclude libcublas* \
   --exclude libnvrtc* \
   --exclude libcuda* \
+  --exclude libcudnn* \
   -w wheel_tmp
 
 
diff --git a/setup.py b/setup.py
@@ -289,7 +289,7 @@ def get_tag(self) -> tuple[str, str, str]:
             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"

Original file line number	Diff line number	Diff line change
`@@ -289,7 +289,7 @@ def get_tag(self) -> tuple[str, str, str]:`
`289`	`289`	`install_requires += [`
`290`	`290`	`"nvidia-cublas-cu12==12.9.*",`
`291`	`291`	`"nvidia-cuda-nvrtc-cu12==12.9.*",`
`292`		`- "nvidia-cudnn-cu12==12.9.*",`
	`292`	`+ "nvidia-cudnn-cu12==9.*",`
`293`	`293`	`]`
`294`	`294`	`else:`
`295`	`295`	`name = "mlx-cpu"`