torch.cuda.get_device_capability¶
- torch.cuda.get_device_capability(device=None)[源代码]¶
获取设备的CUDA能力。
- Parameters
设备 (torch.device 或 整数, 可选) – 返回设备能力的设备。如果此参数为负整数，则此函数不执行任何操作。如果
device为None（默认），则使用当前设备，由current_device()给出。- Returns
设备的CUDA主要和次要能力
- Return type