fix the bug: can not use the CPU to test the model

eacad8cf · huangyuxin · 8432b9b2 · eacad8cf
隐藏空白更改
内联并排

Showing with 7 addition and 3 deletion

deepspeech/utils/log.py deepspeech/utils/log.py +7 -3

未找到文件。
--- 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])
+        if (os.environ['CUDA_VISIBLE_DEVICES'] != ''):
-        infer_config = inference.Config()
+            gpu_id = int(os.environ['CUDA_VISIBLE_DEVICES'].split(',')[0])
-        infer_config.enable_use_gpu(100, gpu_id)
+            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,