老师为什么16g显存的gpu训练这个resnet网络,总是提示coda out of memory.

来源:6-11 PyTorch搭建cifar10训练脚本搭建-Mobilenetv1结构

慕少3205599

2020-07-20

gpu显存不足。

写回答

1回答

会写代码的好厨师

2020-07-23

有具体的log信息么?我看一下log报错信息。按道理不应该的,我的电脑才12g显存都没问题。

0
0

PyTorch入门到进阶 实战计算机视觉与自然语言处理项目

理论基础+技术讲解+实战开发,快速掌握PyTorch框架

1190 学习 · 293 问题

查看课程