From 711d5adb730f59260548e47351719de8e1dbfe3c Mon Sep 17 00:00:00 2001 From: Brandon Thomas Date: Fri, 10 Jun 2022 17:09:33 -0400 Subject: [PATCH] Add architecture debugging --- tts/vocodes_common.py | 2 ++ 1 file changed, 2 insertions(+) diff --git a/tts/vocodes_common.py b/tts/vocodes_common.py index 4b21608..98a1d24 100644 --- a/tts/vocodes_common.py +++ b/tts/vocodes_common.py @@ -35,6 +35,8 @@ def print_gpu_info(): print('PyTorch version', torch.__version__) print('CUDA Available?', torch.cuda.is_available()) print('CUDA Device count', torch.cuda.device_count()) + print('CUDA architectures library was compiled for', torch.cuda.get_arch_list()) + #try: # from tensorflow.python.client import device_lib # print('local devices', str(device_lib.list_local_devices()).replace("\n", "\n "))