Fix GPU installation. Closes #3437

explosion · Mar 19, 2019 · 02d7b41 · 02d7b41
1 parent 9211f30
commit 02d7b41
Showing 1 changed file with 6 additions and 6 deletions.
diff --git a/setup.py b/setup.py
@@ -238,12 +238,12 @@ def setup_package():
             ],
             setup_requires=["wheel"],
             extras_require={
-                "cuda": ["cupy>=4.0"],
-                "cuda80": ["cupy-cuda80>=4.0"],
-                "cuda90": ["cupy-cuda90>=4.0"],
-                "cuda91": ["cupy-cuda91>=4.0"],
-                "cuda92": ["cupy-cuda92>=4.0"],
-                "cuda100": ["cupy-cuda100>=4.0"],
+                "cuda": ["thinc_gpu_ops>=0.0.1,<0.1.0", "cupy>=5.0.0b4"],
+                "cuda80": ["thinc_gpu_ops>=0.0.1,<0.1.0", "cupy-cuda80>=5.0.0b4"],
+                "cuda90": ["thinc_gpu_ops>=0.0.1,<0.1.0", "cupy-cuda90>=5.0.0b4"],
+                "cuda91": ["thinc_gpu_ops>=0.0.1,<0.1.0", "cupy-cuda91>=5.0.0b4"],
+                "cuda92": ["thinc_gpu_ops>=0.0.1,<0.1.0", "cupy-cuda92>=5.0.0b4"],
+                "cuda100": ["thinc_gpu_ops>=0.0.1,<0.1.0", "cupy-cuda100>=5.0.0b4"],
                 # Language tokenizers with external dependencies
                 "ja": ["mecab-python3==0.7"],
             },