Update config.py

2023-05-02 12:31:05 +00:00 · 2023-05-02 12:31:05 +00:00 · c4a18107dc
parent 951989117b
commit c4a18107dc
1 changed files with 1 additions and 1 deletions
--- 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