We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 2ec76a6 commit d2bd709Copy full SHA for d2bd709
lightllm/utils/device_utils.py
@@ -82,7 +82,7 @@ def get_current_device_name():
82
83
if torch.cuda.is_available():
84
device = torch.cuda.current_device()
85
- gpu_name = torch.cuda.get_device_name(device)
+ gpu_name = torch.cuda.get_device_name(device).replace(" ", "_")
86
return gpu_name
87
else:
88
return None
0 commit comments