torch.cuda.get_device_properties

torch.cuda.get_device_properties(device) [source]

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

Параметры:

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

Возвращает:

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

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

_CudaDeviceProperties