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

Workaround bug in thinc require_gpu

This commit is contained in:
Matthew Honnibal 2018-10-14 23:15:08 +02:00
parent 6e6f6be3f5
commit 38aa835ada
1 changed files with 12 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

13
spacy/__init__.py
View File

@ -5,7 +5,7 @@ warnings.filterwarnings("ignore", message="numpy.dtype size changed")
warnings.filterwarnings("ignore", message="numpy.ufunc size changed") warnings.filterwarnings("ignore", message="numpy.ufunc size changed")
# These are imported as part of the API # These are imported as part of the API
from thinc.neural.util import prefer_gpu, require_gpu from thinc.neural.util import prefer_gpu
from .cli.info import info as cli_info from .cli.info import info as cli_info
from .glossary import explain from .glossary import explain
@ -28,3 +28,14 @@ def blank(name, **kwargs):
def info(model=None, markdown=False, silent=False): def info(model=None, markdown=False, silent=False):
return cli_info(model, markdown, silent) return cli_info(model, markdown, silent)
def require_gpu():
from thinc.v2v import Model
from thinc.neural.ops import CupyOps
if CupyOps.xp is None:
raise ValueError(
"GPU is not accessible. Check your LD_LIBRARY_PATH enironment variable "
"and check that thinc was installed with GPU, e.g. thinc[cuda]")
Model.Ops = CupyOps
Model.ops = CupyOps()
return True