从源码看DL4J中Native BLAS的加载,以及配置

望天

发布于 2018-08-02 11:46:25

1.1K0

发布于 2018-08-02 11:46:25

最近在用DeepLearning4J(DL4J)尝试语音识别的深度学习,git DL4J的代码,用IntelliJ IDEA打开,配置好相关依赖后,运行包org.deeplearning4j.examples.test.Test的main,可以正常运行,但是有警告提示如下:

十一月 27, 2015 12:37:07 下午 com.github.fommil.netlib.BLAS <clinit>
WARNING: Failed to load implementation from: com.github.fommil.netlib.NativeSystemBLAS
十一月 27, 2015 12:37:09 下午 com.github.fommil.jni.JniLoader liberalLoad
****************************************************************
INFO: successfully loaded C:\Users\ADMINI~1\AppData\Local\Temp\jniloader6882206374132167742netlib-native_ref-win-x86_64.dll
WARNING: COULD NOT LOAD NATIVE SYSTEM BLAS
ND4J performance WILL be reduced
Please install native BLAS library such as OpenBLAS or IntelMKL
See http://nd4j.org/getstarted.html#open for further details
****************************************************************

提示无法加载com.github.fommil.netlib.NativeSystemBLAS,和无法加载native system blas,DN4J的性能会受到影响.

查了github,stackoverflow,quora等,找到了如下网页.github的是一个遇到类似问题的人抱怨native blas难以配置,害的自己在源码中才找到解决方法,而nd4j的程序员回答所有的深度学习框架中的native blas都是难配置的,我在搜索的时候也发现了MLib等库确实也会报这个警告.第二个博客是一篇讲述如何配置blas的文章.

https://github.com/deeplearning4j/nd4j/issues/456
http://avulanov.blogspot.cz/2014/09/howto-to-run-netlib-javabreeze-in.html

这篇文章中,我想说明一下几个问题:

1. native blas 是本地库,用C/C++写成,因而运算速度较快.

2. Java如果要调用C/C++的dll,一定要用JNI技术来调用对应的dll,那么dll的路径和名称分别是什么?

3. 如果找到需要的dll,假设叫a.dll,如果a.dll又依赖于b.dll和c.dll,那么把a.dll,b.dll,c.dll都放在Java识别的路径下,是不是就可以解决这个问题了.

1.dll的存放路径和名称

先看DL4J的源码,错误首先出现在如下代码中

Nd4j.getRandom().setSeed(seed);

进入Nd4j的类,加断点,单步调试,经过如下的函数调用栈后,进入NativeSystemBLAS这个类.

我们看下NativeSystemBLAS类的内容,在static静态块中找到如下用于加载dll的代码:

static {
        String jnilib = JniNamer.getJniName("netlib-native_system");
        String natives = System.getProperty("com.github.fommil.netlib.NativeSystemBLAS.natives", jnilib);
        JniLoader.load(natives.split(","));
    }

先加断点单步运行,看看jnilib的内容,运行后得到的是"netlib-native_system-win-x86_64.dll",为什么是这个,我的电脑是64位,如果32位系统又会需要哪个dll库呢?

我们进入getJniName函数,看下其代码,这里arch用于获得架构,os用于获得系统版本,extension获得后缀,然后拼出dll的名称:

    public static String getJniName(String stem) {
        String arch = arch();//获得系统架构,x86,i386对应于i686,x86_64,amd64对应于x86_64,其他还有arm等架构
        String abi = abi(arch);//判断是不是arm架构,如果是arm架构,需要其他的设置,这里不问
        String os = os();//判断系统是win,linux.还是mac os x
        String extension = extension(os);//根据系统版本,确定文件后缀名是dll,还是so,osx
        return stem + "-" + os + "-" + arch + abi + "." + extension;//返回确定的库名字
    }

arch代码如下:

private static String arch() {
        String arch = System.getProperty("os.arch", "").toLowerCase();
        if(!arch.equals("x86") && !arch.equals("i386") && !arch.equals("i486") && !arch.equals("i586") && !arch.equals("i686")) {
            if(!arch.equals("x86_64") && !arch.equals("amd64")) {
                if(arch.equals("ia64")) {
                    return "ia64";
                } else if(arch.equals("arm")) {
                    return "arm";
                } else if(arch.equals("armv5l")) {
                    return "armv5l";
                } else if(arch.equals("armv6l")) {
                    return "armv6l";
                } else if(arch.equals("armv7l")) {
                    return "armv7l";
                } else if(arch.equals("sparc")) {
                    return "sparc";
                } else if(arch.equals("sparcv9")) {
                    return "sparcv9";
                } else if(arch.equals("pa_risc2.0")) {
                    return "risc2";
                } else if(arch.equals("ppc")) {
                    return "ppc";
                } else if(arch.startsWith("ppc")) {
                    return "ppc64";
                } else {
                    log.warning("unrecognised architecture: " + arch);
                    return "unknown";
                }
            } else {
                return "x86_64";
            }
        } else {
            return "i686";
        }
    }

abi代码如下:

private static String abi(String arch) {
        if(!arch.startsWith("arm")) {
            return "";
        } else {
            try {
                String[] e = new String[]{"sun.boot.library.path", "java.library.path", "java.home"};
                int len$ = e.length;

                int i$;
                String dir;
                for(i$ = 0; i$ < len$; ++i$) {
                    dir = e[i$];
                    String file = System.getProperty(dir, "");
                    log.config(dir + ": " + file);
                    if(file.matches(".*(gnueabihf|armhf).*")) {
                        return "hf";
                    }
                }

                e = new String[]{"/lib/arm-linux-gnueabihf", "/usr/lib/arm-linux-gnueabihf"};
                len$ = e.length;

                for(i$ = 0; i$ < len$; ++i$) {
                    dir = e[i$];
                    File var7 = new File(dir);
                    if(var7.exists()) {
                        return "hf";
                    }
                }

                return "";
            } catch (SecurityException var6) {
                log.log(Level.WARNING, "unable to detect ABI", var6);
                return "unknown";
            }
        }
    }

os代码如下:

private static String os() {
        String os = System.getProperty("os.name", "").toLowerCase();
        if(os.startsWith("linux")) {
            return "linux";
        } else if(os.startsWith("windows")) {
            return "win";
        } else if(!os.startsWith("mac os x") && !os.startsWith("darwin")) {
            if(os.startsWith("freebsd")) {
                return "freebsd";
            } else if(os.startsWith("android")) {
                return "android";
            } else if(os.startsWith("sunos")) {
                return "sun";
            } else if(os.startsWith("hp-ux")) {
                return "hpux";
            } else if(os.startsWith("kd")) {
                return "kd";
            } else {
                log.warning("unable to detect OS type: " + os);
                return "unknown";
            }
        } else {
            return "osx";
        }
    }

extension代码如下:

 private static String extension(String os) {
        return os.equals("win")?"dll":(os.equals("osx")?"jnilib":"so");
    }

这样,我就知道了dll的具体名字为什么是netlib-native_system-win-x86_64.dll了.

接下来,我还需要知道这个dll要放在哪里,才能被nd4j代码找到!

接下来,我们跳进JniLoader.load(natives.split(",")),看看load是如何执行的:

    public static synchronized void load(String... paths) {
        if(paths != null && paths.length != 0) {
            String[] javaLibPath = paths;
            int arr$ = paths.length;

            int len$;
            String path;
            for(len$ = 0; len$ < arr$; ++len$) {
                String i$ = javaLibPath[len$];
                path = (new File(i$)).getName();
                if(loaded.contains(path)) {
                    log.info("already loaded " + i$);
                    return;
                }
            }

            javaLibPath = System.getProperty("java.library.path").split(File.pathSeparator);//这里获得windows下path对应的路径,是个字符串数组
            String[] var11 = paths;
            len$ = paths.length;

            for(int var12 = 0; var12 < len$; ++var12) {
                path = var11[var12];
                log.config("JNI LIB = " + path);
                String[] extracted = javaLibPath;
                int len$1 = javaLibPath.length;

                for(int i$1 = 0; i$1 < len$1; ++i$1) {
                    String libPath = extracted[i$1];
                    File file = (new File(libPath, path)).getAbsoluteFile();//这里根据path路径和dll的名称拼成一个文件路径,下面检查这个文件是否存在,如果不存在,拼下一个文件路径,如果存在,就加载这个dll
                    log.finest("checking " + file);
                    if(file.exists() && file.isFile() && liberalLoad(file, path)) {
                        return;
                    }
                }

                File var13 = extract(path);
                if(var13 != null && liberalLoad(var13, path)) {
                    return;
                }
            }

            throw new ExceptionInInitializerError("unable to load from " + Arrays.toString(paths));
        } else {
            throw new ExceptionInInitializerError("invalid parameters");
        }
    }

其实这段代码本来我没有看太懂,但是架不住调试功能强大,运行一次,加断点,看看内容怎么变化,就知道是什么意思了.

这里假设我的windows下Path环境变量设置为:"D:\Python\Python35\Scripts\;D:\Program Files\Java\jdk1.7.0_15\bin;C:\WINDOWS;D:\BLAS;",那么javaLibPath的内容就会是这四个组成的字符串数组.接下来取出来第一个,和dll名称(netlib-native_system-win-x86_64.dll)一起拼成一个文件路径"D:\Python\Python35\Scripts\netlib-native_system-win-x86_64.dll",接下来检查这个路径是否是文件,如果不是,就会继续拼下一个,直到拼出"D:\BLAS\netlib-native_system-win-x86_64.dll",然后我的dll文件确实放在这里,程序就会加载.

现在我们找到对应的dll文件,这里为netlib-native_system-win-x86_64.dll,然后放在D:\BLAS\这个位置,把D:\BLAS加入path变量,然后重启Intellij(这一步不一定需要,大家自己尝试),让它重新读取path值.

重新运行下Test里的main函数,然后运行结果竟然还是如上的错误,额,是我们刚才的分析都是错误了么?

2.dll库的依赖和依赖查找

我们在跳进加载dll的地方,看看究竟是哪里错了:

private static boolean liberalLoad(File file, String name) {
        try {
            log.finest("attempting to load " + file);
            System.load(file.getAbsolutePath());
            log.info("successfully loaded " + file);
            loaded.add(name);
            return true;
        } catch (UnsatisfiedLinkError var6) {

            //这里捕获到一个异常,说Can't find dependent libraries.
            log.log(Level.FINE, "skipping load of " + file, var6);
            String tmpdir = System.getProperty("java.io.tmpdir");
            if(tmpdir != null && tmpdir.trim().length() > 2 && file.getAbsolutePath().startsWith(tmpdir)) {
                log.log(Level.FINE, "deleting " + file);

                try {
                    file.delete();
                } catch (Exception var5) {
                    log.info("failed to delete " + file);
                }
            }

            return false;
        } catch (SecurityException var7) {
            log.log(Level.INFO, "skipping load of " + file, var7);
            return false;
        } catch (Throwable var8) {
            throw new ExceptionInInitializerError(var8);
        }
    }

这里我们捕获到一个异常,说 D:\BLAS\netlib-native_system-win-x86_64.dll: Can't find dependent libraries.

原来如此我们的库还需要一些依赖库,这些依赖库没有找到.但是我们怎么知道这个库依赖于哪些库呢?哪些是已经有的,哪些是没有的?这里推荐一个工具,叫PEStudio,可以查看dll文件或exe文件依赖于哪些库:

这里显示了netlib-native_system-win-x86_64.dll依赖于哪些库,其中liblapack3.dll和libblas3.dll是我们需要找到了.其实到这一步,问题就算解决了,参见OpenBlas的官网,这些文件都可以很轻松的下载到,就是要注意首先需要知道你要的是32位还是64位,其次所有dll需要是统一的,不能部分32,部分64. 谢谢曲奇饼的提醒, 这里需要注意的还有jdk的版本, 在64位系统下, 应该使用64的jdk, 大家注意尝试.

这里是我用PEStudio找到的依赖关系:

至此,问题圆满解决.

=========================

最后,把在win10 64位系统下成功运行的库文件放在这:

csdn:http://download.csdn.net/detail/u201011221/9355487

baiduyun: http://pan.baidu.com/s/1jGO5waE

本文参与腾讯云自媒体同步曝光计划，分享自作者个人站点/博客。

原始发表：2015年11月27日，如有侵权请联系 cloudcommunity@tencent.com 删除

其他

本文分享自作者个人站点/博客前往查看

如有侵权，请联系 cloudcommunity@tencent.com 删除。

本文参与腾讯云自媒体同步曝光计划，欢迎热爱写作的你一起参与！

其他

登录后参与评论

0 条评论

热度