mirror of
https://github.com/nomic-ai/gpt4all
synced 2024-11-10 01:10:35 +00:00
python: fix CUDA dependency version (#2858)
Signed-off-by: Jared Van Bortel <jared@nomic.ai>
This commit is contained in:
parent
3ba9c6344d
commit
af9416c0bf
@ -94,8 +94,8 @@ setup(
|
|||||||
],
|
],
|
||||||
extras_require={
|
extras_require={
|
||||||
'cuda': [
|
'cuda': [
|
||||||
'nvidia-cuda-runtime-cu12',
|
'nvidia-cuda-runtime-cu11',
|
||||||
'nvidia-cublas-cu12',
|
'nvidia-cublas-cu11',
|
||||||
],
|
],
|
||||||
'all': [
|
'all': [
|
||||||
'gpt4all[cuda]; platform_system == "Windows" or platform_system == "Linux"',
|
'gpt4all[cuda]; platform_system == "Windows" or platform_system == "Linux"',
|
||||||
|
Loading…
Reference in New Issue
Block a user