print(torch.cuda.memory_summary(), end='')
xiaoxiannvlgy opened this issue · 1 comments
xiaoxiannvlgy commented
AttributeError: module torch.cuda has no attribute memory_summary
What the meaning of "torch.cuda.memory_summary()" ??
What does the author want to express in this sentence “print(torch.cuda.memory_summary(), end='')”
train.py line 100
weiaicunzai commented
Which Pytorch version are you using ?I am using google Colab to train my model which is currently Pytorch 1.6. If you do not want to upgrade your Pytorch, simply comment that line.
You could check the details here: torch.cuda.memory_summary