From be7d057dbcb89abb8b9dc40d366af13df0b0fc48 Mon Sep 17 00:00:00 2001 From: Robert Sachunsky <38561704+bertsky@users.noreply.github.com> Date: Wed, 13 May 2020 13:52:55 +0200 Subject: [PATCH] make CUDA check non-fatal, fix #52 --- ocrd_anybaseocr/cli/ocrd_anybaseocr_block_segmentation.py | 3 +-- 1 file changed, 1 insertion(+), 2 deletions(-) diff --git a/ocrd_anybaseocr/cli/ocrd_anybaseocr_block_segmentation.py b/ocrd_anybaseocr/cli/ocrd_anybaseocr_block_segmentation.py index 896faf4..c497bae 100755 --- a/ocrd_anybaseocr/cli/ocrd_anybaseocr_block_segmentation.py +++ b/ocrd_anybaseocr/cli/ocrd_anybaseocr_block_segmentation.py @@ -75,8 +75,7 @@ def __init__(self, *args, **kwargs): def process(self): if not tf.test.is_gpu_available(): - LOG.error("Your system has no CUDA installed. No GPU detected.") - sys.exit(1) + LOG.warning("Tensorflow cannot detect CUDA installation. Running without GPU will be slow.") try: page_grp, self.image_grp = self.output_file_grp.split(',') except ValueError: