code: def setup_torch_training_env(cudnn_enable, cudnn_benchmark):
      torch.backends.cudnn.enabled = cudnn_enable
      torch.backends.cudnn.benchmark = cudnn_benchmark
      torch.manual_seed(54321)
      use_cuda = torch.cuda.is_available()
      num_gpus = torch.cuda.device_count()
      print(" > Using CUDA: ", use_cuda)
      print(" > Number of GPUs: ", num_gpus)
      return use_cuda, num_gpus
  
file_path: TTS/TTS/utils/training.py
imports: import torch
  import numpy
label_desc: Use image node_id 1 for calling a global function with example usage: setup_torch_training_env(cudnn_enable, cudnn_benchmark) and returns: use_cuda,  num_gpus