caffe:cmake编译指定glog,gflag路径

当使用cmake编译caffe的情况下,在 cmake生成Makefile时会自动找到系统安装的glog,gflag,但是如是我们自己编译了一个glog,gflag,并没有安装在(/usr)系统目录下,而是放在用户目录(/home)下,要想使用这个glog,gflag版本,不做处理cmake是找不到的。 要想在cmake编译caffe时指定glog,gflag路径,需要下面两步:

定义GLOG_ROOT_DIR,GFLAGS_ROOT_DIR参数

#$caffe_root caffe源码根目录 
cmake $caffe_root  -G "Unix Makefiles" \
    -DGLOG_ROOT_DIR=$glog_source_root \
    -DGFLAGS_ROOT_DIR=$gflags_source_root 

这里$glog_source_root,gflags_source_root 是指glog,gflag源码所在的文件夹。

修改FindGlog.cmake,FindGFlags.cmake

$caffe_root/cmake/Modules/下的FindGlog.cmake,FindGFlags.cmake中关于调用find_package寻找glog,gflag的代码有bug(目前还没修改),导致即使如上一步通过GLOG_ROOT_DIR,GFLAGS_ROOT_DIR指定了glog,gflag路径,cmake也不会找到该路径下的glog,gflag,如果你的/usr下安装了glog,gflag,它依然会找到系统路径下的版本。所以要对$caffe_root/cmake/Modules/下的FindGlog.cmake,FindGFlags.cmake做修改: FindGlog.cmake修改如下:

# modified by guyadong
set(GLOG_ROOT_DIR "" CACHE PATH "Folder contains Google glog")
if(WIN32)
    # 增加代码
    find_path(GLOG_INCLUDE_DIR glog/logging.h
        PATHS ${GLOG_ROOT_DIR}/src/windows NO_DEFAULT_PATH)

    find_path(GLOG_INCLUDE_DIR glog/logging.h
        PATHS ${GLOG_ROOT_DIR}/src/windows)
else()
    # 增加代码
    find_path(GLOG_INCLUDE_DIR glog/logging.h
        PATHS ${GLOG_ROOT_DIR}
                NO_DEFAULT_PATH)
    find_path(GLOG_INCLUDE_DIR glog/logging.h
        PATHS ${GLOG_ROOT_DIR})
endif()

if(MSVC)
    find_library(GLOG_LIBRARY_RELEASE libglog_static
        PATHS ${GLOG_ROOT_DIR}
        PATH_SUFFIXES Release NO_DEFAULT_PATH)

    find_library(GLOG_LIBRARY_RELEASE libglog_static
        PATHS ${GLOG_ROOT_DIR}
        PATH_SUFFIXES Release)

    find_library(GLOG_LIBRARY_DEBUG libglog_static
        PATHS ${GLOG_ROOT_DIR}
        PATH_SUFFIXES Debug NO_DEFAULT_PATH)

    find_library(GLOG_LIBRARY_DEBUG libglog_static
        PATHS ${GLOG_ROOT_DIR}
        PATH_SUFFIXES Debug)

    set(GLOG_LIBRARY optimized ${GLOG_LIBRARY_RELEASE} debug ${GLOG_LIBRARY_DEBUG})
else()
    # 增加代码
    find_library(GLOG_LIBRARY glog
        PATHS ${GLOG_ROOT_DIR}
        PATH_SUFFIXES lib lib64
                NO_DEFAULT_PATH)
    find_library(GLOG_LIBRARY glog
        PATHS ${GLOG_ROOT_DIR}
        PATH_SUFFIXES lib lib64)
endif()

FindGFlags.cmake修改如下

# modified by guyadong
# We are testing only a couple of files in the include directories
if(WIN32)
    # 增加代码
    find_path(GFLAGS_INCLUDE_DIR gflags/gflags.h
        PATHS ${GFLAGS_ROOT_DIR}/src/windows NO_DEFAULT_PATH)
    find_path(GFLAGS_INCLUDE_DIR gflags/gflags.h
        PATHS ${GFLAGS_ROOT_DIR}/src/windows)
else()
    # 增加代码
    find_path(GFLAGS_INCLUDE_DIR gflags/gflags.h
        PATHS ${GFLAGS_ROOT_DIR}/include
                NO_DEFAULT_PATH)
    find_path(GFLAGS_INCLUDE_DIR gflags/gflags.h
        PATHS ${GFLAGS_ROOT_DIR}/include)
endif()

if(MSVC)
    # 增加代码
    find_library(GFLAGS_LIBRARY_RELEASE
        NAMES libgflags
        PATHS ${GFLAGS_ROOT_DIR}
        PATH_SUFFIXES Release NO_DEFAULT_PATH)

    find_library(GFLAGS_LIBRARY_RELEASE
        NAMES libgflags
        PATHS ${GFLAGS_ROOT_DIR}
        PATH_SUFFIXES Release)
    # 增加代码
    find_library(GFLAGS_LIBRARY_DEBUG
        NAMES libgflags-debug
        PATHS ${GFLAGS_ROOT_DIR}
        PATH_SUFFIXES Debug NO_DEFAULT_PATH)

    find_library(GFLAGS_LIBRARY_DEBUG
        NAMES libgflags-debug
        PATHS ${GFLAGS_ROOT_DIR}
        PATH_SUFFIXES Debug)

    set(GFLAGS_LIBRARY optimized ${GFLAGS_LIBRARY_RELEASE} debug ${GFLAGS_LIBRARY_DEBUG})
else()
    find_library(GFLAGS_LIBRARY gflags PATHS ${GFLAGS_ROOT_DIR}/lib NO_DEFAULT_PATH)
    find_library(GFLAGS_LIBRARY gflags PATHS ${GFLAGS_ROOT_DIR}/lib)
endif()

比对原始代码 ,就可以发现,只是将每个find_library调用增加了NO_DEFAULT_PATH参数再执行一次

原因分析

这个问题的根本原因在于代码作者对cmake的find_library的查找机制没有全搞明白 简单说就是调用find_library时,会以从一组路径中顺序查找指定的库,这有一个优先序问题,如果没有指定了NO_DEFAULT_PATH,则会先查找默认的系统库路径 如果指定了NO_DEFAULT_PATH,则只查找PATHS提供的路径,不会查找系统库路径。因为所以原始代码中没有加NO_DEFAULT_PATH导致每次只能找到系统路径下的库。 如何保证PATHS指定的路径优先被搜索呢? 解决的办法cmake官网也给出来了,就是调用两次find_library,第一次加NO_DEFAULT_PATH,第二次则不加。

The default search order is designed to be most-specific to least-specific for common use cases. Projects may override the order by simply calling the command multiple times and using the NO_* options: find_library ( NAMES name PATHS paths… NO_DEFAULT_PATH) find_library ( NAMES name) Once one of the calls succeeds the result variable will be set and stored in the cache so that no call will search again.

另外,$caffe_root/cmake/Modules/下的FindLevelDB.cmake,FindLMDB.cmake,FindOpenBLAS.cmake,FindSnappy.cmake都存在相同的问题。解决办法是一样的。

参数资料

《find_library》

本文参与腾讯云自媒体分享计划,欢迎正在阅读的你也加入,一起分享。

发表于

我来说两句

0 条评论
登录 后参与评论

相关文章

来自专栏乐沙弥的世界

Linux下基于源码方式安装MySQL 5.6

    MySQL为开源数据库,因此可以基于源码实现安装。基于源码安装有更多的灵活性。也就是说我们可以针对自己的硬件平台选用合适的编译器来优化编译后的二进制代码...

771
来自专栏Objective-C

React Native 部署开发环境

3835
来自专栏Danny的专栏

java 实现注册时发送激活邮件+激活

版权声明:本文为博主原创文章,未经博主允许不得转载。 https://blog.csdn.net/huyuyang6688/article/...

842
来自专栏高爽的专栏

HTML发送邮件链接

        大家点击我的博客主页详细资料里的邮箱就可以给我发邮件了,那么这个效果是怎么实现的呢?         1. 简单的链接         效果:给...

2520
来自专栏Angular&服务

mac安装mysql

我的博客即将搬运同步至腾讯云+社区,邀请大家一同入驻:https://cloud.tencent.com/developer/support-plan?invi...

963
来自专栏landv

sourcetree跳过注册的方法

1882
来自专栏用户2442861的专栏

ubuntu/win10 CLion 使用

版权声明:本文为博主原创文章,未经博主允许不得转载。 https://blog.csdn.net/haluoluo211/article/details...

832
来自专栏生信技能树

安装VEP及其注释数据库

为了其它相关软件的顺利运行,我们根据教程来设置默认的安装目录及变量环境:Ensembl's VEP , If you don't have VEP instal...

2005
来自专栏闵开慧

修改eclipse中的默认工作路径

首先,关闭Eclipse并打开eclipse的安装目录,依次打开以下目录,configuration\.settings 文件夹下有一个文件 org.eclip...

30610
来自专栏hbbliyong

常用工具整理

代码编辑 Cmd Markdown 免费在线Markdown编辑器,一直再用,很不错。 ? Cmd Markdown下载地址:X86 | X64 反编译...

41810

扫码关注云+社区