From 164cb46fffe0118617deea05d8dcdb0e5e1b425d Mon Sep 17 00:00:00 2001
From: Brandon Thomas <echelon@gmail.com>
Date: Mon, 31 May 2021 23:42:29 -0400
Subject: [PATCH] add debug print to tts inference script

---
 models/tacotron2/inference.py | 7 +++++++
 1 file changed, 7 insertions(+)

diff --git a/models/tacotron2/inference.py b/models/tacotron2/inference.py
index f4423f31e4..4fc646da1b 100755
--- a/models/tacotron2/inference.py
+++ b/models/tacotron2/inference.py
@@ -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")