torch.cuda.change_current_allocator

torch.cuda.change_current_allocator(allocator)

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

如果目前的分配器已經被使用/初始化，此函數將會發生錯誤。

參數

allocator (torch.cuda.memory._CUDAAllocator) – 要設定為作用中的分配器。

注意

請參閱 記憶體管理 以取得建立和使用自定義分配器的詳細資訊