From af9416c0bf4ac445a4ada92d0451c5c8ad18022d Mon Sep 17 00:00:00 2001
From: Jared Van Bortel <jared@nomic.ai>
Date: Tue, 13 Aug 2024 19:11:04 -0400
Subject: [PATCH] python: fix CUDA dependency version (#2858)

Signed-off-by: Jared Van Bortel <jared@nomic.ai>
---
 gpt4all-bindings/python/setup.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/gpt4all-bindings/python/setup.py b/gpt4all-bindings/python/setup.py
index 1374d92d..63bd9140 100644
--- 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"',