評價此頁

torch.cuda.memory.get_allocator_backend#

torch.cuda.memory.get_allocator_backend()[原始碼]#

返回一個字串,描述由 PYTORCH_CUDA_ALLOC_CONF 設定的當前活動的分配器後端。目前可用的後端是 native (PyTorch 的原生快取分配器) 和 cudaMallocAsync` (CUDA 的內建非同步分配器)。

注意

有關選擇分配器後端的詳細資訊,請參閱 記憶體管理

返回型別

str