raise error when run on cpu but set use_gpu as True

PaddlePaddle · Aug 18, 2021 · 30b10d2 · 30b10d2
1 parent 49c4a3a
commit 30b10d2
Showing 1 changed file with 5 additions and 0 deletions.
diff --git a/tools/infer/utility.py b/tools/infer/utility.py
@@ -159,6 +159,11 @@ def create_predictor(args, mode, logger):
  precision = inference.PrecisionType.Float32
 
  if args.use_gpu:
+ gpu_id = get_infer_gpuid()
+ if gpu_id is None:
+ raise ValueError(
+ "Not found GPU in current device. Please check your device or set args.use_gpu as False"
+ )
  config.enable_use_gpu(args.gpu_mem, 0)
  if args.use_tensorrt:
  config.enable_tensorrt_engine(