torch.cuda.memory.get_allocator_backend

torch.cuda.memory.get_allocator_backend() [source]

Возвращает строку с описанием активного бэкенда аллокатора, заданного с помощью PYTORCH_ALLOC_CONF. В настоящее время доступны следующие бэкенды: native (собственный кэширующий аллокатор PyTorch) и cudaMallocAsync` (встроенный асинхронный аллокатор CUDA).

Примечание

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

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

© 2026, PyTorch Contributors
PyTorch has a BSD-style license, as found in the LICENSE file.
https://docs.pytorch.org/docs/2.14/generated/torch.cuda.memory.get_allocator_backend.html