博客
关于我
强烈建议你试试无所不能的chatGPT,快点击我
caffe 加入 cudnn编译
阅读量:4154 次
发布时间:2019-05-25

本文共 2109 字,大约阅读时间需要 7 分钟。

首先要使用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:

转载地址:http://oyeti.baihongyu.com/

你可能感兴趣的文章
阿里p8程序员四年提交6000次代码的确有功,但一次错误让人唏嘘!
查看>>
一道技术问题引起的遐想,最后得出结论技术的本质是多么的朴实!
查看>>
985硕士:非科班自学编程感觉还不如培训班出来的,硕士白读了?
查看>>
你准备写代码到多少岁?程序员们是这么回答的!
查看>>
码农:和产品对一天需求,产品经理的需求是对完了,可我代码呢?
查看>>
程序员过年回家该怎么给亲戚朋友解释自己的职业?
查看>>
技术架构师的日常工作是什么?网友:搭框架,写公共方法?
查看>>
第四章 微信飞机大战
查看>>
九度:题目1008:最短路径问题
查看>>
九度Online Judge
查看>>
九度:题目1027:欧拉回路
查看>>
九度:题目1012:畅通工程
查看>>
九度:题目1017:还是畅通工程
查看>>
九度:题目1034:寻找大富翁
查看>>
第六章 背包问题——01背包
查看>>
51nod 分类
查看>>
1136 . 欧拉函数
查看>>
面试题:强制类型转换
查看>>
Decorator模式
查看>>
Template模式
查看>>