torch.cuda.caching_allocator_delete

torch.cuda.caching_allocator_delete(mem_ptr)

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

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

参数

mem_ptr (int) – 内存地址，由分配器释放。

注意

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