From 41f32653071b25056dca63be1cc012ecfdcd8dbf Mon Sep 17 00:00:00 2001 From: RVC-Boss <129054828+RVC-Boss@users.noreply.github.com> Date: Thu, 24 Aug 2023 21:41:28 +0800 Subject: [PATCH] Update config.py --- config.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/config.py b/config.py index a5a0689..510103b 100644 --- a/config.py +++ b/config.py @@ -92,7 +92,7 @@ class Config: self.gpu_name = torch.cuda.get_device_name(i_device) if ( ("16" in self.gpu_name and "V100" not in self.gpu_name.upper()) - or "P40" in self.gpu_name.upper() + or "P40" in self.gpu_name.upper()or "P10" in self.gpu_name.upper() or "1060" in self.gpu_name or "1070" in self.gpu_name or "1080" in self.gpu_name