mirror of
https://github.com/RVC-Project/Retrieval-based-Voice-Conversion-WebUI.git
synced 2024-12-29 19:15:04 +08:00
Update config.py
This commit is contained in:
parent
951989117b
commit
c4a18107dc
@ -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
|
||||
|
Loading…
Reference in New Issue
Block a user