torch.cuda.get_allocator_backend

torch.cuda.get_allocator_backend()

返回一个字符串，描述由 PYTORCH_CUDA_ALLOC_CONF 设置的活动分配器后端。当前可用的后端有 native （PyTorch 的原生缓存分配器）和 cudaMallocAsync（CUDA 的内置异步分配器）。

注意

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

返回类型:

str