为什么netlib-java native blas / lapack库不能提高性能?

问题描述 投票:5回答:1

我正在使用这段代码来计算火花推荐:

    SparkSession spark = SparkSession
            .builder()
            .appName("SomeAppName")
            .config("spark.master", "local[" + args[2] + "]")
            .config("spark.local.dir",args[4])
            .getOrCreate();
    JavaRDD<Rating> ratingsRDD = spark
            .read().textFile(args[0]).javaRDD()
            .map(Rating::parseRating);
    Dataset<Row> ratings = spark.createDataFrame(ratingsRDD, Rating.class);
    ALS als = new ALS()
            .setMaxIter(Integer.parseInt(args[3]))
            .setRegParam(0.01)
            .setUserCol("userId")
            .setItemCol("movieId")
            .setRatingCol("rating").setImplicitPrefs(true);

    ALSModel model = als.fit(ratings);
    model.setColdStartStrategy("drop");
    Dataset<Row> rowDataset = model.recommendForAllUsers(50);

这些是maven依赖项,使这段代码工作:

    <dependency>
        <groupId>org.apache.spark</groupId>
        <artifactId>spark-core_2.11</artifactId>
        <version>2.4.0</version>
    </dependency>
    <dependency>
        <groupId>org.apache.spark</groupId>
        <artifactId>spark-sql_2.11</artifactId>
        <version>2.4.0</version>
    </dependency>
    <dependency>
        <groupId>org.apache.spark</groupId>
        <artifactId>spark-mllib_2.11</artifactId>
        <version>2.4.0</version>
    </dependency>
    <dependency>
        <groupId>org.scala-lang</groupId>
        <artifactId>scala-library</artifactId>
        <version>2.11.8</version>
    </dependency>

使用此代码计算建议需要大约70秒的数据文件。此代码生成以下警告:

WARN BLAS: Failed to load implementation from: com.github.fommil.netlib.NativeSystemBLAS
WARN BLAS: Failed to load implementation from: com.github.fommil.netlib.NativeRefBLAS
WARN LAPACK: Failed to load implementation from: com.github.fommil.netlib.NativeSystemLAPACK
WARN LAPACK: Failed to load implementation from: com.github.fommil.netlib.NativeRefLAPACK

现在我尝试通过在maven中添加此依赖项来启用netlib-java:

    <dependency>
        <groupId>com.github.fommil.netlib</groupId>
        <artifactId>all</artifactId>
        <version>1.1.2</version>
        <type>pom</type>
    </dependency>

为了避免这个新环境的崩溃我不得不做这个额外的伎俩:

LD_PRELOAD=/usr/lib64/libopenblas.so

现在它也可以工作,它不会发出任何警告,但它的工作速度较慢,平均需要约170秒来执行相同的计算。我在CentOS上运行它。

本机库不应该更快吗?有可能让它更快吗?

maven apache-spark-mllib recommender-systems netlib-java
1个回答
0
投票
  1. 首先你可以检查你的centos版本,因为centos 6可能不使用本机库,检查this
  2. 据我所知,ALS算法自2.0版本以来得到了改进,你可以查看Highlights in 2.2 而来自2.2的源代码就是打击: enter image description here 所以本地图书馆没有帮助!
© www.soinside.com 2019 - 2024. All rights reserved.