torch.cuda.memory.change_current_allocator

torch.cuda.memory.change_current_allocator(allocator) [исходный код]

Изменяет используемый в данный момент распределитель памяти на указанный.

Если текущий распределитель уже использовался или был инициализирован, эта функция вызовет ошибку.

Параметры:

allocator (torch.cuda.memory._CUDAAllocator) – распределитель, который будет установлен в качестве активного.

Примечание

Подробнее о создании и использовании пользовательского распределителя см. в разделе Управление памятью.