mirror of
https://github.com/storytold/artcraft.git
synced 2026-11-02 16:38:14 +00:00
don't print devices
This commit is contained in:
1 parent
6516a04eda
commit
e4fe7f8ed1
1 file changed
+6
-6
@@ -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):
|
||||
|
||||
Reference in new issue
Block a user