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