torch.mps.recommended_max_memory

torch.mps.recommended_max_memory() [source]

Возвращает рекомендуемый максимальный размер рабочего набора для памяти GPU в байтах.

Примечание

Рекомендуемый максимальный размер рабочего набора для Metal возвращается из device.recommendedMaxWorkingSetSize.

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

int