From 44ae5543e0c4476f7c5324a6fdb74c066fdbde30 Mon Sep 17 00:00:00 2001
From: Justin John <34035011+justinjohn0306@users.noreply.github.com>
Date: Thu, 11 Jan 2024 10:29:54 +0530
Subject: [PATCH] Update styletts2importable.py

---
 tts/StyleTTS2/styletts2importable.py | 31 ++++++++++++++++++++++++++++
 1 file changed, 31 insertions(+)

diff --git a/tts/StyleTTS2/styletts2importable.py b/tts/StyleTTS2/styletts2importable.py
index 5e7b14b..e31df16 100644
--- a/tts/StyleTTS2/styletts2importable.py
+++ b/tts/StyleTTS2/styletts2importable.py
@@ -1,3 +1,34 @@
+import torch
+from time import  strftime
+import os, sys, time
+import platform
+
+print("Env vars:")
+print(os.environ)
+
+
+def print_gpu_info():
+    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('CUDA threads for parallelizing CPU operations', torch.get_num_threads())
+    print('CUDA architectures library was compiled for', torch.cuda.get_arch_list())
+
+    #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)
+
+
+print_gpu_info(
+
+)
+
 print("NLTK")
 import nltk
 nltk.download('punkt')