初学深度学习,想用VGG模型做些案例,自己的笔记本显卡1060 6G,跑代码时候会出错,查了下有说显卡内存不够的,想问下大佬们用类似VGG这些模型时候,设备都是什么样的,1060 能跑动VGG 模型吗?万分感谢~
1060也不是不可以跑,需要把batch搞小一些,多加pooling,缩小模型参数规模。当然会影响性能和精度,但是这个不可避免。在笔记本上跑,比较容易烧毁显卡,那样损失就大了。建议你买个二手矿卡的1070Ti 8G,不到2000的价格,接近1080的性能。性价比超级高。