explosion/spaCy
1
1
Fork 0
mirror of https://github.com/explosion/spaCy.git synced 2024-12-26 01:46:28 +03:00
Code Issues Packages Projects Releases Wiki Activity

Unhack prefer_gpu

Browse Source
This commit is contained in:
Matthew Honnibal 2018-10-14 23:27:09 +02:00
parent 2ad3a4ea32
commit 8ccfa52d19
1 changed files with 1 additions and 12 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 from thinc.neural.util import prefer_gpu, require_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,14 +28,3 @@ 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