Merge pull request explosion#10849 from danieldk/simplify-gpu-check

danieldk · web-flow · commit 09a5f03dd795 · 2022-05-30T16:35:10.000+02:00
Simplify GPU check
diff --git a/spacy/cli/_util.py b/spacy/cli/_util.py
@@ -12,7 +12,7 @@
 from typer.main import get_command
 from contextlib import contextmanager
 from thinc.api import Config, ConfigValidationError, require_gpu
-from thinc.util import has_cupy, gpu_is_available
+from thinc.util import gpu_is_available
 from configparser import InterpolationError
 import os
 
@@ -554,5 +554,5 @@ def setup_gpu(use_gpu: int, silent=None) -> None:
         require_gpu(use_gpu)
     else:
         local_msg.info("Using CPU")
-        if has_cupy and gpu_is_available():
+        if gpu_is_available():
             local_msg.info("To switch to GPU 0, use the option: --gpu-id 0")
