PyTorch+Spark

spark-sklearn

Github仓库已经归档了，推荐使用Joblib Apache Spark Backend

安装

官方文档

pip install SciPy

apt-get install g++

pip install spark-sklearn

参考链接

将pytorch 模型嵌入到spark中进行大规模预测