diff --git a/deepspeech/utils/log.py b/deepspeech/utils/log.py index 065a4c84d3fad900f9354e1f55539a33f8f5e216..3fd7d24800383d287ad5c32f142cb7e0489fcc0e 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)