storytold/storyteller-ml
2
0
Fork 0
mirror of https://github.com/storytold/storyteller-ml.git synced 2026-10-28 09:08:54 +00:00
Code Issues Packages Projects Releases Wiki Activity

Add architecture debugging

Browse Source
This commit is contained in:
Brandon Thomas
2022-06-10 17:09:33 -04:00
parent 44f2a1082b
commit 711d5adb73
1 changed files with 2 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files
tts/vocodes_common.py
+2
View File
@@ -35,6 +35,8 @@ def print_gpu_info():
print('PyTorch version', torch.__version__) print('PyTorch version', torch.__version__)
print('CUDA Available?', torch.cuda.is_available()) print('CUDA Available?', torch.cuda.is_available())
print('CUDA Device count', torch.cuda.device_count()) print('CUDA Device count', torch.cuda.device_count())
print('CUDA architectures library was compiled for', torch.cuda.get_arch_list())
#try: #try:
# from tensorflow.python.client import device_lib # from tensorflow.python.client import device_lib
# print('local devices', str(device_lib.list_local_devices()).replace("\n", "\n ")) # print('local devices', str(device_lib.list_local_devices()).replace("\n", "\n "))