torch.cuda.get_allocator_backend

torch.cuda.get_allocator_backend()

傳回一個字串，描述由 PYTORCH_CUDA_ALLOC_CONF 設定的活動記憶體配置器後端。 目前可用的後端有 native (PyTorch 的原生快取記憶體配置器) 和 cudaMallocAsync` (CUDA 的內建非同步記憶體配置器)。

注意

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

回傳型別

str