github/gpt4all
1
0
Fork 0
mirror of https://github.com/nomic-ai/gpt4all.git synced 2025-09-04 10:04:35 +00:00
Code Issues Packages Projects Releases Wiki Activity

python: depend on offical NVIDIA CUDA packages (#2355)

Browse Source 
Signed-off-by: Jared Van Bortel <jared@nomic.ai>
This commit is contained in:
Jared Van Bortel
2024-05-20 18:06:27 -04:00
committed by GitHub
parent c779d8a32d
commit 09dd3dc318
2 changed files with 39 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
gpt4all-bindings/python/setup.py
View File

@@ -93,7 +93,15 @@ setup(
'typing-extensions>=4.3.0; python_version >= "3.9" and python_version < "3.11"',
],
extras_require={
'cuda': [
'nvidia-cuda-runtime-cu12',
'nvidia-cublas-cu12',
],
'all': [
'gpt4all[cuda]; platform_system == "Windows" or platform_system == "Linux"',
],
'dev': [
'gpt4all[all]',
'pytest',
'twine',
'wheel',