[misc] add cuda runtime version to usage data (#12190)

Signed-off-by: youkaichao <youkaichao@gmail.com>
Co-authored-by: Roger Wang <ywang@roblox.com>
This commit is contained in:
youkaichao
2025-01-21 08:31:01 +08:00
committed by GitHub
parent af69a6aded
commit da7512215f

View File

@@ -130,6 +130,7 @@ class UsageMessage:
self.total_memory: Optional[int] = None
self.architecture: Optional[str] = None
self.platform: Optional[str] = None
self.cuda_runtime: Optional[str] = None
self.gpu_count: Optional[int] = None
self.gpu_type: Optional[str] = None
self.gpu_memory_per_device: Optional[int] = None
@@ -169,6 +170,8 @@ class UsageMessage:
self.gpu_count = torch.cuda.device_count()
self.gpu_type = device_property.name
self.gpu_memory_per_device = device_property.total_memory
if current_platform.is_cuda():
self.cuda_runtime = torch.version.cuda
self.provider = _detect_cloud_provider()
self.architecture = platform.machine()
self.platform = platform.platform()