Jupyter运行显存爆炸,明明上一个单元格已经运行完毕为什么还是会炸?

问题再现

上一个单元格运行完了train(),我想要用模型输出做点东西,可是提醒我显存不够;
在命令行中查看显存占用情况,发现4张卡都占满了,可真是太厉害了!

解决方案

查看原来写的validate(),发现其中有意识地不使用tensor而是使用numpy,所以改用numpy操作,并且使用清除目前不需要的显存语句;

问题得到解决,下一步弄清楚为什么即使单元格运行停止了,显存还会占用这么大

你可能感兴趣的:(#,读论文,写代码,jupyter,ide,python)