From 19dc42776af33ee209c256c739303f27aa458144 Mon Sep 17 00:00:00 2001 From: Adriane Boyd Date: Fri, 24 Jul 2020 09:26:26 +0200 Subject: [PATCH] Remove hard-coded GPU ID from pretrain (#5808) --- spacy/cli/pretrain.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/spacy/cli/pretrain.py b/spacy/cli/pretrain.py index 6d6c65161..e949f76cf 100644 --- a/spacy/cli/pretrain.py +++ b/spacy/cli/pretrain.py @@ -125,7 +125,7 @@ def pretrain( config[key] = str(config[key]) util.fix_random_seed(seed) - has_gpu = prefer_gpu(gpu_id=1) + has_gpu = prefer_gpu() msg.info("Using GPU" if has_gpu else "Not using GPU") output_dir = Path(output_dir)