This commit is contained in:
LDOUBLEV 2021-11-09 12:16:32 +08:00
parent b1d5e43ce1
commit c8cf31236f

View File

@ -180,7 +180,7 @@ def create_predictor(args, mode, logger):
gpu_id = get_infer_gpuid() gpu_id = get_infer_gpuid()
if gpu_id is None: if gpu_id is None:
logger.warning( logger.warning(
"GPU is not found in current device by nvidia-smi. Please check your device or ignored it if run on jeston." "GPU is not found in current device by nvidia-smi. Please check your device or ignore it if run on jeston."
) )
config.enable_use_gpu(args.gpu_mem, 0) config.enable_use_gpu(args.gpu_mem, 0)
if args.use_tensorrt: if args.use_tensorrt: