From 6516a04eda91ed8b3ef13e77740b32f356f18fa2 Mon Sep 17 00:00:00 2001 From: Brandon Thomas Date: Sun, 22 Aug 2021 19:41:50 -0400 Subject: [PATCH] print devices --- models/tacotron2/vocodes_common.py | 6 ++++++ 1 file changed, 6 insertions(+) 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):