torch.cuda.change_current_allocator

torch.cuda.change_current_allocator(allocator) [source]

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

Параметры

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

Примечание

См. Управление памятью для получения подробной информации о создании и использовании пользовательского менеджера памяти