Catch RuntimeError in torch.cuda.reset_peak_memory_stats - Githubissues

Stonesjtu / pytorch_memlab

Profiling and inspecting memory in pytorch

MIT License

1.01k stars 37 forks source link

Catch RuntimeError in torch.cuda.reset_peak_memory_stats #26

Closed hauntsaninja closed 3 years ago

hauntsaninja commented 3 years ago

The AttributeError was fixed in https://github.com/pytorch/pytorch/pull/48406

Stonesjtu commented 3 years ago

lgtm。 Thanks for the PR!