RuntimeError:cuda runtime error(2):out of memory
lizxchen opened this issue · comments
有两个8G的显卡，显示这个错误，想问一下原因，谢谢
把 config.py 中的 batch_size 改小
chenyuntc / pytorch-best-practice
A Guidance on PyTorch Coding Style Based on Kaggle Dogs vs. Cats
A Guidance on PyTorch Coding Style Based on Kaggle Dogs vs. Cats
lizxchen opened this issue · comments
有两个8G的显卡，显示这个错误，想问一下原因，谢谢
把 config.py 中的 batch_size 改小