红联Linux门户
Linux帮助

torch/caffe等框架下解决内存无法garbage collection的问题

发布时间:2016-09-07 15:17:42来源:linux网站作者:Alanyannick
1.安装libjemalloc-dev 
命令为:apt-get install libjemalloc-dev 
然后在需要运行的环境中输入: (preload so operation)
LD_PRELOAD=/usr/lib/x86_64-linux-gnu/libjemalloc.so xxx 
eg. LD_PRELOAD=/usr/lib/x86_64-linux-gnu/libjemalloc.so th train.lua 
 
2.所有的程序都使用这个libjemalloc作为后台内存管理,则将其设为环境变量中
~/.bachrc 添加:
export LD_PRELOAD=$LD_PRELOAD:/usr/lib/x86_64-linux-gnu/libjemalloc.so 
然后:
source ~/.bachrc
 
本文永久更新地址:http://www.linuxdiyf.com/linux/23960.html