首页 > 其他 > 详细

caffe 配置 札记

时间:2016-01-22 17:39:09      阅读:167      评论:0      收藏:0      [点我收藏+]

技术分享

cudnn的配置

1. 安装前请去先官网下载cuDNN (cudnn-7.0-linux-x64-v3),建议安装v3,v4有些问题

将cudnn-7.0-linux-x64-v3解压后会有两个文件夹include和lib64,复制其中文件到系统库

sudo cp include/cudnn.h /usr/local/include
sudo cp lib64/libcudnn.* /usr/local/lib

2. 链接cuDNN的库文件

sudo ln -sf /usr/local/lib/libcudnn.so.7.0.64 /usr/local/lib/libcudnn.so.7.0
sudo ln -sf /usr/local/lib/libcudnn.so.7.0 /usr/local/lib/libcudnn.so
sudo ldconfig -v

3. 在Makefile.config中启用CUDNN,去掉注释

USE_CUDNN := 1

参考资料

caffe 配置 札记

原文:http://www.cnblogs.com/vin-yuan/p/5151716.html

(0)
(0)
   
举报
评论 一句话评论(0
关于我们 - 联系我们 - 留言反馈 - 联系我们:wmxa8@hotmail.com
© 2014 bubuko.com 版权所有
打开技术之扣,分享程序人生!