caffe 加入 cudnn编译

Posted 2022-12-13 sunmenggmail

首先要使用cudnn v2，

ALL PLATFORMS


    Extract the cuDNN archive to a directory of your choice, referred to below as <installpath>.
    Then follow the platform-specific instructions as follows.


LINUX


    cd <installpath>
    export LD_LIBRARY_PATH=`pwd`:$LD_LIBRARY_PATH


    Add <installpath> to your build and link process by adding -I<installpath> to your compile
    line and -L<installpath> -lcudnn to your link line.

注意的是

在Makefile中新添的语句要在之前的前面

LDFLAGS += -L/home/research/sunmeng/software/cudnn-6.5-linux-x64-v2 -lcudnn
LDFLAGS += $(foreach librarydir,$(LIBRARY_DIRS),-L$(librarydir)) $(PKG_CONFIG) \\
         $(foreach library,$(LIBRARIES),-l$(library))

//

遇到的一个问题是把caffe编译为一个动态链接库，其他程序调用caffe.so时，也需要加载cudnn.so，则编译选项添加 -L/usr/local/cuda/lib64 -L/home/xxxxx/cudnn-6.5-linux-x64-v2/ 

一直提示是undefined reference to `cudnnGetConvolutionForwardWorkspaceSize' 之类的错误，这是因为在动态链接库中找不到相应的函数实现

这是因为之前在 /usr/local/cuda/lib64目录下拷贝了一个老版本的cudnn.so, 而我们想要使用的是新版本的/home/xxxxx/cudnn-6.5-linux-x64-v2，g++先在lib64找，找到了就不在后面的目录找了，解决这个问题很简单，就是把两个目录的顺序调换即可

///

程序在运行时就要需要加载动态链接库了，原理参考

http://blog.sina.com.cn/s/blog_484102dd0101114f.html

所以安装共享库后要注意共享库路径设置问题, 如下:

1) 如果共享库文件安装到了/lib或/usr/lib目录下, 那么需执行一下ldconfig命令

ldconfig命令的用途, 主要是在默认搜寻目录(/lib和/usr/lib)以及动态库配置文件/etc/ld.so.conf内所列的目录下, 搜索出可共享的动态链接库(格式如lib*.so*), 进而创建出动态装入程序(ld.so)所需的连接和缓存文件. 缓存文件默认为/etc/ld.so.cache, 此文件保存已排好序的动态链接库名字列表. 

2) 如果共享库文件安装到了/usr/local/lib(很多开源的共享库都会安装到该目录下)或其它"非/lib或/usr/lib"目录下, 那么在执行ldconfig命令前, 还要把新共享库目录加入到共享库配置文件/etc/ld.so.conf中, 如下:

# cat /etc/ld.so.conf
include ld.so.conf.d/*.conf
# echo "/usr/local/lib" >> /etc/ld.so.conf
# ldconfig

3) 如果共享库文件安装到了其它"非/lib或/usr/lib" 目录下,  但是又不想在/etc/ld.so.conf中加路径(或者是没有权限加路径). 那可以export一个全局变量LD_LIBRARY_PATH, 然后运行程序的时候就会去这个目录中找共享库. 

LD_LIBRARY_PATH的意思是告诉loader在哪些目录中可以找到共享库. 可以设置多个搜索目录, 这些目录之间用冒号分隔开. 比如安装了一个mysql到/usr/local/mysql目录下, 其中有一大堆库文件在/usr/local/mysql/lib下面, 则可以在.bashrc或.bash_profile或shell里加入以下语句即可:

export LD_LIBRARY_PATH=/usr/local/mysql/lib:$LD_LIBRARY_PATH    

一般来讲这只是一种临时的解决方案, 在没有权限或临时需要的时候使用.

当然也可以将动态链接库的地址写入到可执行文件中去，这样在运行程序的时候就不用考虑动态链接库的地址了，注意，最好是绝对路径，就是在编译中添加 

-Wl,-rpath,/home/xxxx/cudnn-6.5-linux-x64-v2:/usr/local/cuda/lib64: