From fb98862337144590caf325abf48e80f390f3244b Mon Sep 17 00:00:00 2001 From: Adriane Boyd Date: Tue, 2 Mar 2021 15:11:18 +0100 Subject: [PATCH] Add hint for --gpu-id to CLI device info (#7234) * Add hint for --gpu-id to CLI device info If the user has `cupy` and an available GPU, add a hint about using `--gpu-id 0` to the CLI output. * Undo change to original CPU message --- spacy/cli/_util.py | 3 +++ 1 file changed, 3 insertions(+) diff --git a/spacy/cli/_util.py b/spacy/cli/_util.py index 86b3ab356..228cc622a 100644 --- a/spacy/cli/_util.py +++ b/spacy/cli/_util.py @@ -11,6 +11,7 @@ from click.parser import split_arg_string 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 configparser import InterpolationError import os @@ -510,3 +511,5 @@ def setup_gpu(use_gpu: int) -> None: require_gpu(use_gpu) else: msg.info("Using CPU") + if has_cupy and gpu_is_available(): + msg.info("To switch to GPU 0, use the option: --gpu-id 0")