快捷方式

torch.cuda.caching_allocator_delete

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

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

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

参数

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

注意

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

文档

访问 PyTorch 的全面开发者文档

查看文档

教程

获得针对初学者和高级开发人员的深入教程

查看教程

资源

查找开发资源并获得问题的解答

查看资源