您的位置：首页 > Web前端

caffe 配置 札记

2016-01-22 16:41 218 查看



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


参考资料

参考资料2
内容来自用户分享和网络整理，不保证内容的准确性，如有侵权内容，可联系管理员处理 点击这里给我发消息
标签： 
相关文章推荐
新的分享
章节导航