Spark 上的 Tensorflow Java API 在 new Graph() 上失败



我正在运行Tensorflow的Java API(版本1.8(来评估Spark上已经训练好的模型。但是,当代码到达此命令Graph gr = new Graph()代码失败时,大多数情况下,有时它会通过。

这发生在执行器和驱动程序上,即使Graph gr= new Graph()是我保留的唯一 TensorFlow 命令。使用 Try-Catch 没有用,因为当它到达此行时,执行程序将失败并退出。

知道可能是什么问题吗?

我们想出了问题所在: tensorflow 的 Java API 依赖于 Visual C++ 重新分发。当我们在Spark上运行它时,一些机器会错过依赖项*.dlls。这会导致 JVM 崩溃。

最新更新