上一个单元格运行完了train(),我想要用模型输出做点东西,可是提醒我显存不够; 在命令行中查看显存占用情况,发现4张卡都占满了,可真是太厉害了!
train()
查看原来写的validate(),发现其中有意识地不使用tensor而是使用numpy,所以改用numpy操作,并且使用清除目前不需要的显存语句;
validate()
tensor
numpy
问题得到解决,下一步弄清楚为什么即使单元格运行停止了,显存还会占用这么大
京公网安备 11010502049817号