python: fix CUDA dependency version (#2858)

Signed-off-by: Jared Van Bortel <jared@nomic.ai>
1 month ago · af9416c0bf
parent 3ba9c6344d
commit af9416c0bf
1 changed files with 2 additions and 2 deletions
--- a/gpt4all-bindings/python/setup.py
+++ b/gpt4all-bindings/python/setup.py
@ -94,8 +94,8 @@ setup(
    ],
    extras_require={
        'cuda': [
-            'nvidia-cuda-runtime-cu12',
-            'nvidia-cublas-cu12',
+            'nvidia-cuda-runtime-cu11',
+            'nvidia-cublas-cu11',
        ],
        'all': [
            'gpt4all[cuda]; platform_system == "Windows" or platform_system == "Linux"',