torch.cuda.get_device_properties

torch.cuda.get_device_properties(device=None) [исходный код]

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

Параметры:

device (torch.device or int or str, необязательно) – устройство, свойства которого нужно вернуть. Если device равно None (по умолчанию), используются свойства текущего устройства, определяемого функцией current_device().

Возвращает:

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

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

_CudaDeviceProperties