做网站那个公司,wordpress注册弹出框,上海 网站开发 兼职,360建筑网电脑版问题再现
上一个单元格运行完了train()#xff0c;我想要用模型输出做点东西#xff0c;可是提醒我显存不够#xff1b; 在命令行中查看显存占用情况#xff0c;发现4张卡都占满了#xff0c;可真是太厉害了#xff01;
解决方案
查看原来写的validate()#xff0c;发…问题再现
上一个单元格运行完了train()我想要用模型输出做点东西可是提醒我显存不够 在命令行中查看显存占用情况发现4张卡都占满了可真是太厉害了
解决方案
查看原来写的validate()发现其中有意识地不使用tensor而是使用numpy所以改用numpy操作并且使用清除目前不需要的显存语句
问题得到解决下一步弄清楚为什么即使单元格运行停止了显存还会占用这么大
