評價此頁

torch.accelerator.memory.empty_cache#

torch.accelerator.memory.empty_cache()[原始碼]#

釋放快取分配器當前佔用的所有未使用的快取記憶體,以便它們可供其他應用程式使用。

注意

如果當前 accelerator 的記憶體分配器尚未初始化,則此函式將不執行任何操作。