torch.cuda.caching_allocator_delete

torch.cuda.caching_allocator_delete(mem_ptr)[源代码]

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

使用 caching_allocator_alloc() 分配的内存在这里被释放。分配器会跟踪相关的设备和流。

参数

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

注意

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