Skip to content

Commit d2bd709

Browse files
committed
fix
1 parent 2ec76a6 commit d2bd709

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

lightllm/utils/device_utils.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -82,7 +82,7 @@ def get_current_device_name():
8282

8383
if torch.cuda.is_available():
8484
device = torch.cuda.current_device()
85-
gpu_name = torch.cuda.get_device_name(device)
85+
gpu_name = torch.cuda.get_device_name(device).replace(" ", "_")
8686
return gpu_name
8787
else:
8888
return None

0 commit comments

Comments
 (0)