torch.cuda.memory.caching_allocator_delete

torch.cuda.memory.caching_allocator_delete(mem_ptr)

使用 CUDA 記憶體分配器分配的記憶體會被刪除。

使用 caching_allocator_alloc() 分配的記憶體在此處釋放。關聯的裝置和流在分配器內部跟蹤。

引數

mem_ptr (int) – 要由分配器釋放的記憶體地址。

注意

有關 GPU 記憶體管理的更多詳細資訊，請參閱 記憶體管理。