torch.cuda.memory.change_current_allocator

torch.cuda.memory.change_current_allocator(allocator)

將當前使用的記憶體分配器更改為提供的分配器。

如果當前分配器已被使用/初始化，此函式將報錯。

引數

allocator (torch.cuda.memory._CUDAAllocator) – 將設定為活動狀態的分配器。

注意

有關建立和使用自定義分配器的詳細資訊，請參閱 記憶體管理。