torch.cuda.get_device_capability — PyTorch 2.7 documentation (original) (raw)

• Docs >
• torch.cuda >
• torch.cuda.get_device_capability
• 

Shortcuts

torch.cuda.get_device_capability(device=None)[source][source]¶

Get the cuda capability of a device.

Parameters

device (torch.device or int or str, optional) – device for which to return the device capability. This function is a no-op if this argument is a negative integer. It uses the current device, given bycurrent_device(), if device is None(default).

Returns

the major and minor cuda capability of the device

Return type

tuple(int, int)

• torch.cuda.get_device_capability
  • get_device_capability()

To analyze traffic and optimize your experience, we serve cookies on this site. By clicking or navigating, you agree to allow our usage of cookies. As the current maintainers of this site, Facebook’s Cookies Policy applies. Learn more, including about available controls: Cookies Policy.