torch.cuda.memory_summary()
can give KeyError
#117130
Labels
module: cuda
Related to torch.cuda, and CUDA support in general
triaged
This issue has been looked at a team member, and triaged and prioritized into an appropriate module
🐛 Describe the bug
Calling
torch.cuda.memory_summary()
can give aKeyError
under certain circumstances.The root cause here is that
torch.cuda.memory_stats()
(which is used internally bytorch.cuda.memory_summary()
) has returned an object that does not have the expected keys.Versions
cc @ptrblck
The text was updated successfully, but these errors were encountered: