Understanding CUDA Memory Usage

若API未标明支持情况则代表该API暂无验证结果，待验证后更新表格。

API名称

是否支持

限制与说明

torch.cuda.memory._record_memory_history

      

torch.cuda.memory._snapshot

      

torch.cuda.memory._dump_snapshot

      
父主题： PyTorch原生API