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

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

Browse Source
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',
'dill>=0.2,<0.3'],
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=[
'Development Status :: 5 - Production/Stable',
'Environment :: Console',