torch.cuda.get_allocator_backend

torch.cuda.get_allocator_backend() [source]

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

Примечание

Для получения подробной информации о выборе бэкенда аллокатора см. Управление памятью.

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

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