From eacad8cf600d2c1d2c6d57ae4ebe3c9c39b434d3 Mon Sep 17 00:00:00 2001 From: huangyuxin Date: Mon, 2 Aug 2021 07:12:10 +0000 Subject: [PATCH] fix the bug: can not use the CPU to test the model --- deepspeech/utils/log.py | 10 +++++++--- 1 file changed, 7 insertions(+), 3 deletions(-) diff --git a/deepspeech/utils/log.py b/deepspeech/utils/log.py index e99dacec..065a4c84 100644 --- a/deepspeech/utils/log.py +++ b/deepspeech/utils/log.py @@ -157,9 +157,13 @@ class Autolog: model_precision="fp32"): import auto_log pid = os.getpid() - gpu_id = int(os.environ['CUDA_VISIBLE_DEVICES'].split(',')[0]) - infer_config = inference.Config() - infer_config.enable_use_gpu(100, gpu_id) + if (os.environ['CUDA_VISIBLE_DEVICES'] != ''): + gpu_id = int(os.environ['CUDA_VISIBLE_DEVICES'].split(',')[0]) + infer_config = inference.Config() + infer_config.enable_use_gpu(100, gpu_id) + else: + gpu_id = None + infer_config = inference.Config() autolog = auto_log.AutoLogger( model_name=model_name, model_precision=model_precision,