码迷,mamicode.com
首页 > 其他好文 > 详细

caffe 配置 札记

时间:2016-01-22 17:39:25      阅读:200      评论: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
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!