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()
if gpu_id is None:
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)
if args.use_tensorrt: