From 604dc01456f351c97ef8a1ae3091192a3395d7ef Mon Sep 17 00:00:00 2001 From: huangyuxin Date: Mon, 2 Aug 2021 07:12:10 +0000 Subject: [PATCH 1/2] 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 e99dacece..065a4c84d 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, From 8a531f7040d88ad801e8e5d0a0697330d7f09c4b Mon Sep 17 00:00:00 2001 From: huangyuxin Date: Mon, 2 Aug 2021 07:33:30 +0000 Subject: [PATCH 2/2] add strip for CUDA_VISIBLE_DEVICES --- deepspeech/utils/log.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/deepspeech/utils/log.py b/deepspeech/utils/log.py index 065a4c84d..3fd7d2480 100644 --- a/deepspeech/utils/log.py +++ b/deepspeech/utils/log.py @@ -157,7 +157,7 @@ class Autolog: model_precision="fp32"): import auto_log pid = os.getpid() - if (os.environ['CUDA_VISIBLE_DEVICES'] != ''): + if (os.environ['CUDA_VISIBLE_DEVICES'].strip() != ''): gpu_id = int(os.environ['CUDA_VISIBLE_DEVICES'].split(',')[0]) infer_config = inference.Config() infer_config.enable_use_gpu(100, gpu_id)