add strip for CUDA_VISIBLE_DEVICES

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

Showing with 1 addition and 1 deletion

deepspeech/utils/log.py deepspeech/utils/log.py +1 -1

未找到文件。
--- 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)