首页 文章 精选 留言 我的

精选列表

搜索[Bug修复],共10000篇文章
优秀的个人博客,低调大师

tensorflow+java 内存泄漏修复

前段时间java程序,内存泄漏比较严重,平均3-5天就要重启一下,赶快分析原因。从公司的监控看到,主要是对外内存泄漏,因为堆内存泄漏不是很严重。所以决定优先处理前者。因为该项目是java开发的,主要任务时加载tensorflow1.*的模型,并实时预测。其实主要JNI调用c++接口,所以很大可能是在接口调用时泄漏了,看代码: Tensorres=null; try{ List<String>colname=IntColname; Runnerrlt=sess.runner(); for(inti=0;i<intvalue[0].length;i++){ int[][]index=newint[intvalue.length][1]; for(intm=0;m<intvalue.length;m++){ index[m][0]=intvalue[m][i]; } TensorindexTensor=Tensor.create(index); rlt.feed(colname.get(i),indexTensor); } colname=FloatColname; for(inti=0;i<floatvalue[0].length;i++){ float[][]index=newfloat[floatvalue.length][1]; for(intm=0;m<floatvalue.length;m++){ index[m][0]=floatvalue[m][i]; } TensorindexTensor=Tensor.create(index); rlt.feed(colname.get(i),indexTensor); temp.add(indexTensor); } res=rlt.fetch("output").run().get(0); float[][]finalRlt=newfloat[intvalue.length][1]; res.copyTo(finalRlt); List<Float>result=newArrayList<Float>(); for(inti=0;i<finalRlt.length;i++){ result.add(finalRlt[i][0]); } returnresult; }catch(Exceptione){ logger.error("",e); }finally{ if(res!=null){ res.close(); } } 虽然res调用了close方法,但是indexTensor 却没有调用,因此调整代码。 Tensorres=null; List<Tensor>temp=newArrayList<>(); try{ List<String>colname=IntColname; Runnerrlt=sess.runner(); for(inti=0;i<intvalue[0].length;i++){ int[][]index=newint[intvalue.length][1]; for(intm=0;m<intvalue.length;m++){ index[m][0]=intvalue[m][i]; } TensorindexTensor=Tensor.create(index); rlt.feed(colname.get(i),indexTensor); temp.add(indexTensor); } colname=FloatColname; for(inti=0;i<floatvalue[0].length;i++){ float[][]index=newfloat[floatvalue.length][1]; for(intm=0;m<floatvalue.length;m++){ index[m][0]=floatvalue[m][i]; } TensorindexTensor=Tensor.create(index); rlt.feed(colname.get(i),indexTensor); temp.add(indexTensor); } res=rlt.fetch("output").run().get(0); float[][]finalRlt=newfloat[intvalue.length][1]; res.copyTo(finalRlt); List<Float>result=newArrayList<Float>(); for(inti=0;i<finalRlt.length;i++){ result.add(finalRlt[i][0]); } returnresult; }catch(Exceptione){ logger.error("",e); }finally{ if(res!=null){ res.close(); } for(inti=0;i<temp.size();i++){ Tensort=temp.get(i); if(t!=null){ t.close(); } } } 主要是增加temp队列回收临时变量。上线后发现泄漏不是很严重了,说明很有效果。不过还有,后面是java堆内存泄漏,发现每次泄漏的时间正是模型切换的时间,因此大概率是模型切换的代码有问题,上代码。 publicstaticvoidclearCache(){ logger.info("===modelclear==="); sess=null; } 这里的session就是模型解析好的session会话,由下面的代码生成。 byte[]graphBytes=IOUtils.toByteArray(inputStream); Graphgraph=newGraph(); graph.importGraphDef(graphBytes); sess=newSession(graph); 立刻明白,上面的代码有问题,调整为如下: publicstaticvoidclearCache(){ logger.info("===modelclear==="); if(sess!=null){ logger.info("destorysession"); sess.close(); sess=null; } } 少了一个close,造成session关联的对象无法释放,至此内存泄漏问题算是解决了。

资源下载

更多资源
Nacos

Nacos

Nacos /nɑ:kəʊs/ 是 Dynamic Naming and Configuration Service 的首字母简称,一个易于构建 AI Agent 应用的动态服务发现、配置管理和AI智能体管理平台。Nacos 致力于帮助您发现、配置和管理微服务及AI智能体应用。Nacos 提供了一组简单易用的特性集,帮助您快速实现动态服务发现、服务配置、服务元数据、流量管理。Nacos 帮助您更敏捷和容易地构建、交付和管理微服务平台。

Rocky Linux

Rocky Linux

Rocky Linux(中文名:洛基)是由Gregory Kurtzer于2020年12月发起的企业级Linux发行版,作为CentOS稳定版停止维护后与RHEL(Red Hat Enterprise Linux)完全兼容的开源替代方案,由社区拥有并管理,支持x86_64、aarch64等架构。其通过重新编译RHEL源代码提供长期稳定性,采用模块化包装和SELinux安全架构,默认包含GNOME桌面环境及XFS文件系统,支持十年生命周期更新。

Sublime Text

Sublime Text

Sublime Text具有漂亮的用户界面和强大的功能,例如代码缩略图,Python的插件,代码段等。还可自定义键绑定,菜单和工具栏。Sublime Text 的主要功能包括:拼写检查,书签,完整的 Python API , Goto 功能,即时项目切换,多选择,多窗口等等。Sublime Text 是一个跨平台的编辑器,同时支持Windows、Linux、Mac OS X等操作系统。

WebStorm

WebStorm

WebStorm 是jetbrains公司旗下一款JavaScript 开发工具。目前已经被广大中国JS开发者誉为“Web前端开发神器”、“最强大的HTML5编辑器”、“最智能的JavaScript IDE”等。与IntelliJ IDEA同源,继承了IntelliJ IDEA强大的JS部分的功能。

用户登录
用户注册