mirror of
https://github.com/storytold/artcraft.git
synced 2026-11-05 17:58:55 +00:00
add debug print to tts inference script
This commit is contained in:
1 parent
446983a477
commit
164cb46fff
1 file changed
+7
@@ -19,6 +19,13 @@ from train import load_model
|
||||
from text import text_to_sequence
|
||||
#from denoiser import Denoiser
|
||||
|
||||
print('========================================')
|
||||
print('Python interpreter', sys.executable)
|
||||
print('PyTorch version', torch.__version__)
|
||||
print('CUDA Available?', torch.cuda.is_available())
|
||||
print('CUDA Device count', torch.cuda.device_count())
|
||||
print('========================================', flush=True)
|
||||
|
||||
# NB(bt, 2021-05-31): Trying to get everything on the same device
|
||||
torch.device("cuda" if torch.cuda.is_available() else "cpu")
|
||||
|
||||
|
||||
Reference in new issue
Block a user