评价此页

torch.cuda.memory.get_allocator_backend#

torch.cuda.memory.get_allocator_backend()[源代码]#

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

注意

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

返回类型

str