From e4fe7f8ed129dbcdd274d2d0503ef28f9a6ca400 Mon Sep 17 00:00:00 2001 From: Brandon Thomas Date: Sun, 22 Aug 2021 19:42:07 -0400 Subject: [PATCH] don't print devices --- models/tacotron2/vocodes_common.py | 12 ++++++------ 1 file changed, 6 insertions(+), 6 deletions(-) diff --git a/models/tacotron2/vocodes_common.py b/models/tacotron2/vocodes_common.py index 992e5b4d33..af3b251a62 100644 --- a/models/tacotron2/vocodes_common.py +++ b/models/tacotron2/vocodes_common.py @@ -26,12 +26,12 @@ def print_gpu_info(): print('PyTorch version', torch.__version__) print('CUDA Available?', torch.cuda.is_available()) print('CUDA Device count', torch.cuda.device_count()) - try: - from tensorflow.python.client import device_lib - print('local devices', str(device_lib.list_local_devices()).replace("\n", "\n ")) - except ImportError: - print('no tensorflow - cannot list devices') - pass + #try: + # from tensorflow.python.client import device_lib + # print('local devices', str(device_lib.list_local_devices()).replace("\n", "\n ")) + #except ImportError: + # print('no tensorflow - cannot list devices') + # pass print('========================================', flush=True) def load_tacotron_model(checkpoint_path):