spark2调用TensorFlow2模型

本文主要是介绍spark2调用TensorFlow2模型，希望对大家解决编程问题提供一定的参考价值，需要的开发者们随着小编来一起学习吧！

问题一：
com.google.protobuf.Parser.parseFrom方法找不到。因cdh默认的spark依赖jar中protobuf-java-[version].jar版本太低，需要手动升级替换；

问题二：
序列化org.tensorflow.SavedModelBundle后广播各种空指针，建议直接改道addFile(modelPath,true)于executor加载模型；
sparkConf.set("spark.serializer", "org.apache.spark.serializer.KryoSerializer")
sparkConf.registerKryoClasses(Array(classOf[org.tensorflow.SavedModelBundle]))

参考：
https://blog.csdn.net/zimiao552147572/article/details/105330740#comments_16159981
https://github.com/tensorflow/java/issues/298
https://www.jianshu.com/p/6442eec4a567
https://www.iteblog.com/archives/1704.html
https://blog.csdn.net/yisun123456/article/details/107688948

这篇关于spark2调用TensorFlow2模型的文章就介绍到这儿，希望我们推荐的文章对编程师们有所帮助！

---