ND4J与Flink集成时ProcessFunction内存泄漏问题求助

• 强制显式释放ND4J堆外内存
  ND4J的ndarray默认占用堆外内存，即便Java层对象被GC，堆外内存未必会立刻回收。在每轮计算结束后，手动调用ndarray.close()或Nd4j.getMemoryManager().release(ndarray)，确保堆外内存被主动释放。如果用了Workspace，要确认在scope结束时调用Workspace.close()，排查是否有嵌套Workspace未正确关闭的情况。

• 开启ND4J内存调试定位泄漏点
  启动时添加系统参数-Dorg.nd4j.debug.memory=true和-Dorg.nd4j.debug.memory.log=true，ND4J会打印每个ndarray的分配、释放日志，直接定位未被释放的内存块。另外可以在计算前后调用Nd4j.getMemoryManager().printMemoryStats()，对比内存分配和释放的字节数，确认泄漏的具体环节。

• 适配Flink TaskManager内存配置与ND4J策略
  在flink-conf.yaml中调大taskmanager.memory.off-heap.size，确保ND4J的堆外内存使用被Flink管控，避免超出系统可用内存导致内存无法回收。同时添加参数-Dorg.nd4j.disable.cache=true，禁用ND4J的内存缓存，防止未使用的内存块被持续占用。

• 排查隐式引用导致的内存无法回收
  检查ProcessFunction里是否有静态集合、ThreadLocal变量无意中持有ndarray引用，导致对象无法被GC回收。另外要确认异步计算逻辑（比如Future回调）有没有保留ndarray的引用，延长了对象生命周期。

• 升级ND4J到最新稳定版
  部分旧版ND4J存在已知的内存泄漏bug，比如某些线性代数操作未正确释放临时内存块，升级到1.0.0-M2.1及以上的最新稳定版，大概率能修复这类问题。

内容的提问来源于stack exchange，提问作者Corvus