From 8ccfa52d19362d172ecb2f2accc76021729ac6da Mon Sep 17 00:00:00 2001 From: Matthew Honnibal Date: Sun, 14 Oct 2018 23:27:09 +0200 Subject: [PATCH] Unhack prefer_gpu --- spacy/__init__.py | 13 +------------ 1 file changed, 1 insertion(+), 12 deletions(-) diff --git a/spacy/__init__.py b/spacy/__init__.py index 80e66023c..411883eae 100644 --- a/spacy/__init__.py +++ b/spacy/__init__.py @@ -5,7 +5,7 @@ warnings.filterwarnings("ignore", message="numpy.dtype size changed") warnings.filterwarnings("ignore", message="numpy.ufunc size changed") # These are imported as part of the API -from thinc.neural.util import prefer_gpu +from thinc.neural.util import prefer_gpu, require_gpu from .cli.info import info as cli_info from .glossary import explain @@ -28,14 +28,3 @@ def blank(name, **kwargs): def info(model=None, markdown=False, silent=False): 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