C 知道
试了几个办法比如把批量大小从128改到32,换更小的数据集都是显示这个要用3.5g内存,这内存到底和啥有关啊,问了chatgpt哪个方法还能用啊
收起
是 GPU 内存不够吧?如果是训练的话,4G内存的显卡太小了。
行吧
报告相同问题?
程序员都在用的中文IT技术交流社区
专业的中文 IT 技术社区,与千万技术人共成长
关注【CSDN】视频号,行业资讯、技术分享精彩不断,直播好礼送不停!