评价此页

torch.cuda.memory.get_allocator_backend#

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

返回一个描述活动分配器后端的字符串,该后端由 PYTORCH_CUDA_ALLOC_CONF 设置。当前可用的后端包括 native(PyTorch 原生的缓存分配器)和 cudaMallocAsync`(CUDA 内置的异步分配器)。

注意

有关选择分配器后端的详细信息,请参阅 内存管理

返回类型

str