Update utils.py

ktransformers/util/utils.py

@@ -33,6 +33,18 @@ import socket
 
 warm_uped = False
 
+def get_device_name(device:torch.device = None):
+    if torch.cuda.is_available():
+        if device is None:
+            num_gpus = torch.cuda.device_count()
+            gpu_name = []
+            for gpu_id in range(num_gpus):
+                gpu_name.append(torch.cuda.get_device_name(gpu_id))
+            return gpu_name
+        else:
+            return torch.cuda.get_device_name(device)
+
+
 def get_free_ports(n: int, continue_prot: list):
     sockets = []
     ports = []