Rooba
/
spaCy
mirror of https://github.com/explosion/spaCy.git
1
1
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Make thinc look in /usr/local/cuda for cuda by default

This commit is contained in:
Matthew Honnibal 2018-05-19 18:12:23 +02:00
parent b9e415a5f8
commit 260707a4c3
1 changed files with 6 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
setup.py
View File

@ -199,6 +199,12 @@ def setup_package():
'regex==2017.4.5', 'regex==2017.4.5',
'dill>=0.2,<0.3'], 'dill>=0.2,<0.3'],
setup_requires=['wheel'], setup_requires=['wheel'],
extras_require={
'cuda': ['cupy>=4.0'],
'cuda80': ['cupy-cuda80>=4.0'],
'cuda90': ['cupy-cuda90>=4.0'],
'cuda91': ['cupy-cuda91>=4.0'],
},
classifiers=[ classifiers=[
'Development Status :: 5 - Production/Stable', 'Development Status :: 5 - Production/Stable',
'Environment :: Console', 'Environment :: Console',