Fix docker image to run on my system's host CUDA version

voice_conversion/softvc/cli_inference.py

@@ -13,6 +13,10 @@ import sys
 import torch
 import torchaudio
 
+from gpu_utils import print_gpu_info
+
+print_gpu_info()
+
 from collections import OrderedDict
 from time import time
 from torch.nn.modules.utils import consume_prefix_in_state_dict_if_present
@@ -28,12 +32,9 @@ sys.path.append("./hifigan/")
 from hifigan.generator import HifiganGenerator
 from hifigan.generator import consume_prefix_in_state_dict_if_present
 
-from gpu_utils import print_gpu_info
 
 # ===========================================
 
-print_gpu_info()
-
 parser = argparse.ArgumentParser()
 parser.add_argument('--acoustic_model_filename', type=str, required=True)
 parser.add_argument('--hifigan_model_filename', type=str, required=True)