python: fix CUDA dependency version (#2858)

Signed-off-by: Jared Van Bortel <jared@nomic.ai>
This commit is contained in:
Jared Van Bortel 2024-08-13 19:11:04 -04:00 committed by GitHub
parent 3ba9c6344d
commit af9416c0bf
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194

View File

@ -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"',