From 97b51f275fa047a52c83755e87c4d6558c376aed Mon Sep 17 00:00:00 2001 From: Kevin Malana Date: Thu, 16 Apr 2026 10:02:33 +0000 Subject: [PATCH] fix(testing_utils): guard get_device_capability with torch.cuda.is_available() Fixes crash in get_device_properties() when CUDA is installed but no GPU is present (e.g., CPU-only cloud studio with CUDA libraries). The function checked IS_CUDA_SYSTEM (torch.version.cuda is not None) but then called torch.cuda.get_device_capability() without verifying an actual GPU exists. Fixes #45341 --- src/transformers/testing_utils.py | 2 ++ 1 file changed, 2 insertions(+) diff --git a/src/transformers/testing_utils.py b/src/transformers/testing_utils.py index 863242a695c6..6e70be2798ba 100644 --- a/src/transformers/testing_utils.py +++ b/src/transformers/testing_utils.py @@ -3207,6 +3207,8 @@ def get_device_properties() -> DeviceProperties: if IS_CUDA_SYSTEM or IS_ROCM_SYSTEM: import torch + if not torch.cuda.is_available(): + return (torch_device, None, None) major, minor = torch.cuda.get_device_capability() if IS_ROCM_SYSTEM: return ("rocm", major, minor)