include cudnn as python dep

2025-12-16 01:49:05 +08:00 · 2025-07-18 06:54:41 -07:00
parent 180ec0d3a5
commit 75bcb46069
2 changed files with 2 additions and 1 deletions
--- a/setup.py
+++ b/setup.py
@@ -289,6 +289,7 @@ if __name__ == "__main__":
            install_requires += [
                "nvidia-cublas-cu12==12.9.*",
                "nvidia-cuda-nvrtc-cu12==12.9.*",
+                "nvidia-cudnn-cu12==12.9.*",
            ]
        else:
            name = "mlx-cpu"