diff --git a/voice_conversion/so-vits-svc-paul/Dockerfile b/voice_conversion/so-vits-svc-paul/Dockerfile index 67065d4..b4370e3 100644 --- a/voice_conversion/so-vits-svc-paul/Dockerfile +++ b/voice_conversion/so-vits-svc-paul/Dockerfile @@ -32,8 +32,7 @@ COPY requirements-without-poetry.txt . #RUN pip install -r requirements-without-poetry.txt -# NB: --copies flag creates a "thick" venv that copies the system python binary -RUN python3 -m venv --copies python \ +RUN python3 -m venv python \ && . python/bin/activate \ && pip install -r requirements-without-poetry.txt diff --git a/voice_conversion/so-vits-svc-paul/src/so_vits_svc_fork/fakeyou_infer.py b/voice_conversion/so-vits-svc-paul/src/so_vits_svc_fork/fakeyou_infer.py index 9acf6fc..57d428c 100644 --- a/voice_conversion/so-vits-svc-paul/src/so_vits_svc_fork/fakeyou_infer.py +++ b/voice_conversion/so-vits-svc-paul/src/so_vits_svc_fork/fakeyou_infer.py @@ -25,6 +25,25 @@ from rich.logging import RichHandler 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 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() + def init_logger() -> None: IN_COLAB = os.getenv("COLAB_RELEASE_TAG") IS_TEST = "test" in Path(__file__).parent.stem