torch.cuda.get_device_properties

torch.cuda.get_device_properties(device) [source]

Получает свойства устройства.

Параметры

device (torch.device или int или str) – устройство, для которого требуется вернуть свойства.

Возвращает

свойства устройства

Тип возвращаемого значения

_CudaDeviceProperties