PR #45351: fix(testing_utils): guard get_device_capability with torch.cuda.is_available()

• Repository: huggingface/transformers
• Author: RudrenduPaul
• State: open | merged: False
• Link: https://github.com/huggingface/transformers/pull/45351

Description (problem / solution / changelog)

What does this PR do?

Fixes a crash in get_device_properties() in testing_utils.py when CUDA is installed on the system but no GPU device is present (e.g., a CPU-only cloud studio with CUDA libraries installed).

The function called torch.cuda.get_device_capability() immediately after checking IS_CUDA_SYSTEM (which is True whenever torch.version.cuda is not None), without first verifying that an actual GPU is available. On CUDA-installed but GPU-less systems, get_device_capability() raises an error.

Fixes #45341

Changes

• src/transformers/testing_utils.py: Add if not torch.cuda.is_available(): return (torch_device, None, None) guard inside the IS_CUDA_SYSTEM or IS_ROCM_SYSTEM branch of get_device_properties(), before the get_device_capability() call.

Tests

This is a fix to the test infrastructure itself (testing_utils.py). The change prevents a crash that occurs in environments where IS_CUDA_SYSTEM=True but no physical GPU is present (e.g., running pytest on a CPU-only Lightning AI studio).

No new tests were added because the existing test suite runs in environments where torch.cuda.is_available() is True — the crash scenario only reproduces on CUDA-installed, no-GPU systems.

Note: This PR was developed with AI assistance (Claude Code). I have reviewed every line and understand the change. This is not a duplicate of any existing open PR (checked open PRs searching for issue 45341 in body and keyword searches for get_device_capability + is_available).

Changed files

• src/transformers/testing_utils.py (modified, +2/-0)