torch.cuda.caching_allocator_delete

torch.cuda.caching_allocator_delete(mem_ptr)[source][source]

删除使用 CUDA 内存分配器分配的内存。

使用 caching_allocator_alloc() .分配的内存在此处释放。关联的设备和流在分配器内部跟踪。

参数:

mem_ptr（int）- 分配器将要释放的内存地址。

注意

更多关于 GPU 内存管理的详细信息，请参阅内存管理。