如图,在跑完训练集以后,显存占用和训练时一样,一直保持七点多,然后去跑测试集,结果提示内存不足。
测试集没有求导,而且在测试集运行释放内存的函数也没有用,显存占用还是那么多。
需要把释放内存的函数放在运行训练集的cell跑吗?

如图,在跑完训练集以后,显存占用和训练时一样,一直保持七点多,然后去跑测试集,结果提示内存不足。
测试集没有求导,而且在测试集运行释放内存的函数也没有用,显存占用还是那么多。
需要把释放内存的函数放在运行训练集的cell跑吗?

你这个是普通家用卡,不推荐,建议用专业卡,专业卡通常有48GB或者96GB的内存,而你这种家庭游戏卡,只有几GB