From 02d7b41893a98800c3a9514234a1d525868e611e Mon Sep 17 00:00:00 2001 From: Matthew Honnibal Date: Wed, 20 Mar 2019 00:59:27 +0100 Subject: [PATCH] Fix GPU installation. Closes #3437 --- setup.py | 12 ++++++------ 1 file changed, 6 insertions(+), 6 deletions(-) diff --git a/setup.py b/setup.py index 34c92ad2b..6f29e1efa 100755 --- 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"], },