From c4a18107dca74e74c2284e04d3f8a791a1476029 Mon Sep 17 00:00:00 2001 From: RVC-Boss <129054828+RVC-Boss@users.noreply.github.com> Date: Tue, 2 May 2023 12:31:05 +0000 Subject: [PATCH] Update config.py --- config.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/config.py b/config.py index 5fc5d62..75ac4ce 100644 --- a/config.py +++ b/config.py @@ -53,7 +53,7 @@ class Config: i_device = int(self.device.split(":")[-1]) self.gpu_name = torch.cuda.get_device_name(i_device) if ( - ("16" in gpu_name and "V100" not in gpu_name.upper()) + ("16" in self.gpu_name and "V100" not in self.gpu_name.upper()) or "P40" in self.gpu_name.upper() or "1070" in self.gpu_name or "1080" in self.gpu_name