深度学习VGG模型加载硬件条件

初学深度学习,想用VGG模型做些案例,自己的笔记本显卡1060 6G,跑代码时候会出
错,查了下有说显卡内存不够的,想问下大佬们用类似VGG这些模型时候,设备都是什
么样的,1060 能跑动VGG 模型吗?
万分感谢~

0

1个回答

1060也不是不可以跑,需要把batch搞小一些,多加pooling,缩小模型参数规模。当然会影响性能和精度,但是这个不可避免。
在笔记本上跑,比较容易烧毁显卡,那样损失就大了。建议你买个二手矿卡的1070Ti 8G,不到2000的价格,接近1080的性能。性价比超级高。

0
Csdn user default icon
上传中...
上传图片
插入图片
抄袭、复制答案,以达到刷声望分或其他目的的行为,在CSDN问答是严格禁止的,一经发现立刻封号。是时候展现真正的技术了!