Add device check in check_gpu

PaddlePaddle · Nov 29, 2023 · 6ddac52 · 6ddac52
1 parent a0901d2
commit 6ddac52
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tools/infer/utility.py b/tools/infer/utility.py
@@ -675,7 +675,7 @@ def get_minarea_rect_crop(img, points):
 
 
 def check_gpu(use_gpu):
-    if use_gpu and not paddle.is_compiled_with_cuda():
+    if use_gpu and (not paddle.is_compiled_with_cuda() or paddle.device.get_device() == 'cpu'):
         use_gpu = False
     return use_gpu