API接口 |
说明 |
|---|---|
torch_npu.npu.caching_allocator_alloc |
Torch_npu提供内存管理相关的部分接口，具体可参考torch.cuda。 |
torch_npu.npu.caching_allocator_delete |
|
torch_npu.npu.set_per_process_memory_fraction |
|
torch_npu.npu.empty_cache |
|
torch_npu.npu.memory_stats |
|
torch_npu.npu.memory_stats_as_nested_dict |
|
torch_npu.npu.reset_accumulated_memory_stats |
|
torch_npu.npu.reset_peak_memory_stats |
|
torch_npu.npu.reset_max_memory_allocated |
|
torch_npu.npu.reset_max_memory_cached |
|
torch_npu.npu.memory_allocated |
|
torch_npu.npu.max_memory_allocated |
|
torch_npu.npu.memory_reserved |
|
torch_npu.npu.max_memory_reserved |
|
torch_npu.npu.memory_cached |
|
torch_npu.npu.max_memory_cached |
|
torch_npu.npu.memory_snapshot |
|
torch_npu.npu.memory_summary |