don't print devices

This commit is contained in:
Brandon Thomas committed 2021-08-22 19:42:07 -04:00
1 parent 6516a04eda
commit e4fe7f8ed1
1 file changed
+6 -6
+6 -6
View File
@@ -26,12 +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
#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):