Merge pull request #58 from bertsky/patch-1

make CUDA check non-fatal, fix #52
OCR-D · May 13, 2020 · 122a848 · 122a848
2 parents e5601f2 + be7d057
commit 122a848
Showing 1 changed file with 1 addition and 2 deletions.
diff --git 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: