diff --git a/models/tacotron2/vocodes_common.py b/models/tacotron2/vocodes_common.py index 13960873e0..992e5b4d33 100644 --- a/models/tacotron2/vocodes_common.py +++ b/models/tacotron2/vocodes_common.py @@ -26,6 +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 print('========================================', flush=True) def load_tacotron_model(checkpoint_path):