Skip to content

GitLab

Unverified Commit ab16f2e4 authored by dyning's avatar dyning Committed by GitHub
Browse files

Merge pull request #3728 from LDOUBLEV/fix_tests 

raise error when run on cpu but set use_gpu as True
parents d1642ed5 30b10d29
Hide whitespace changes
Inline Side-by-side
Showing with 5 additions and 0 deletions
tools/infer/utility.py
View file @ ab16f2e4
...@@ -159,6 +159,11 @@ def create_predictor(args, mode, logger): ...@@ -159,6 +159,11 @@ def create_predictor(args, mode, logger):
precision = inference.PrecisionType.Float32 precision = inference.PrecisionType.Float32
if args.use_gpu: 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) config.enable_use_gpu(args.gpu_mem, 0)
if args.use_tensorrt: if args.use_tensorrt:
config.enable_tensorrt_engine( config.enable_tensorrt_engine(
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment