print devices

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):