diff --git a/mlx/version.h b/mlx/version.h index 7adac11607..94b181cec7 100644 --- 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 index 60e2ab11e6..e83fc4406f 100644 --- 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 index 8a55171126..81a06c3a26 100644 --- a/setup.py +++ b/setup.py @@ -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"