explosion/spaCy
1
1
Fork 0
mirror of https://github.com/explosion/spaCy.git synced 2024-12-26 09:56:28 +03:00
Code Issues Packages Projects Releases Wiki Activity

Fix specifiers for GPU

Browse Source
This commit is contained in:
Matthew Honnibal 2018-10-15 00:08:44 +02:00
parent f02bb08f39
commit 7202abdfa9
1 changed files with 6 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

12
setup.py
View File

@ -200,12 +200,12 @@ def setup_package():
'requests>=2.13.0,<3.0.0',
'pathlib==1.0.1; python_version < "3.4"'],
extras_require={
'cuda': ['thinc[cuda]>=6.10.6,<6.11.0'],
'cuda80': ['thinc[cuda80]>=6.10.6,<6.11.0'],
'cuda90': ['thinc[cuda91]>=6.10.6,<6.11.0'],
'cuda91': ['thinc[cuda91]>=6.10.6,<6.11.0'],
'cuda92': ['thinc[cuda92]>=6.10.6,<6.11.0'],
'cuda10': ['thinc[cuda100]>=6.10.6,<6.11.0'],
'cuda': ['cupy>=4.0'],
'cuda80': ['cupy-cuda80>=4.0', 'thinc_gpu_ops>=0.0.3,<0.1.0'],
'cuda90': ['cupy-cuda90>=4.0', 'thinc_gpu_ops>=0.0.3,<0.1.0'],
'cuda91': ['cupy-cuda91>=4.0', 'thinc_gpu_ops>=0.0.3,<0.1.0'],
'cuda92': ['cupy-cuda92>=4.0', 'thinc_gpu_ops>=0.0.3,<0.1.0'],
'cuda100': ['cupy-cuda100>=4.0', 'thinc_gpu_ops>=0.0.3,<0.1.0']
},
classifiers=[
'Development Status :: 5 - Production/Stable',