Understanding CUDA Memory Usage

若API未标明支持情况，则代表该API的支持情况待验证。

API名称

是否支持

限制与说明

torch.cuda.memory._record_memory_history

      

torch.cuda.memory._snapshot

      

torch.cuda.memory._dump_snapshot

      
父主题： PyTorch原生API