From 260707a4c3d5919b7b97e59520d42a58a5d6ceb8 Mon Sep 17 00:00:00 2001
From: Matthew Honnibal <honnibal+gh@gmail.com>
Date: Sat, 19 May 2018 18:12:23 +0200
Subject: [PATCH] Make thinc look in /usr/local/cuda for cuda by default

---
 setup.py | 6 ++++++
 1 file changed, 6 insertions(+)

diff --git a/setup.py b/setup.py
index 739bcfdc3..2dec51f8a 100755
--- a/setup.py
+++ b/setup.py
@@ -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',