文章详情

短信预约-IT技能 免费直播动态提醒

请输入下面的图形验证码

提交验证

短信预约提醒成功

Linux下怎么安装卷积神经网络框架caffe

2023-06-27 14:16

关注

这篇文章主要介绍了Linux下怎么安装卷积神经网络框架caffe的相关知识,内容详细易懂,操作简单快捷,具有一定借鉴价值,相信大家阅读完这篇Linux下怎么安装卷积神经网络框架caffe文章都会有所收获,下面我们一起来看看吧。

caffe是一个清晰而高效的深度学习框架,是纯粹的C++/CUDA架构,支持命令行、python和matlab接口;可以在CPU和GPU之间无缝切换,其作者是贾扬清,目前caffe2已经发布。

Linux下怎么安装卷积神经网络框架caffe

下载caffe并保存到一个目录下(推荐放到 /home 目录)

  1. 安装依赖项:
sudo apt-get install libprotobuf-dev libleveldb-dev libsnappy-dev libopencv-dev libhdf5-serial-dev protobuf-compiler(这一大串是一个命令)sudo apt-get install --no-install-recommends libboost-all-devsudo apt-get install libatlas-base-devsudo apt-get install python-devsudo apt-get install libgflags-dev libgoogle-glog-dev liblmdb-devsudo apt-get install python-lmdbsudo apt-get install python-pip1234567
  1. cd caffe,并修改配置文件
cp Makefile.config.example Makefile.config(如果用CPU跑,把这两个文件的cuda相关都注释掉)1
  1. 修改需编译的选项(针对ubuntu15.04以及之后的版本,之前的版本不用修改,用于解决无法找到hdf5.h等文件)
INCLUDE_DIRS := $(PYTHON_INCLUDE) /usr/local/include \  /usr/lib/x86_64-linux-gnu/hdf5/serial/includeLIBRARY_DIRS := $(PYTHON_LIB) /usr/local/lib /usr/lib \ /usr/lib/x86_64-linux-gnu/hdf5/serial123456

如果第4步make runtest有错误,error while loading shared libraries: libcudart.so.8.0: cannot open shared object file: can not open(/usr/local/lib/目录下找不到libcudart.so.8.0这个文件, 该文件可以在usr/local/cuda-8.0/lib64/目录下找到),解决办法:

sudo cp /usr/local/cuda-8.0/lib64/libcudart.so.8.0 /usr/local/lib/libcudart.so.8.0 && sudo ldconfigsudo cp /usr/local/cuda-8.0/lib64/libcublas.so.8.0 /usr/local/lib/libcublas.so.8.0 && sudo ldconfig  sudo cp /usr/local/cuda-8.0/lib64/libcurand.so.8.0 /usr/local/lib/libcurand.so.8.0 && sudo ldconfigsudo cp /usr/local/cuda-8.0/lib64/libcudnn.so.6 /usr/local/lib/libcudnn.so.6 && sudo ldconfig1234

或者可以在LIBRARY_DIRS后添加这个路径?usr/local/cuda-8.0/lib64/(没试过,应该可以)

参考链接:https://blog.csdn.net/qq_38451119/article/details/81126692

去掉CPU—only选项前的注释,使之生效,若需Python layer,则还需去掉WITH_PYTHON_LAYER :=1前的注释 如需编译MATLAB接口,则需修改MATLAB_DIR,改成当前系统下的MATLAB路径

  1. 编译caffe:
make all -j4make test -j4make runtest123

编译python接口

sudo pip install -r python/requirements.txtmake pycaffe12

编译matlab接口

make matcaffegedit ~/.bashrc12

最后面添加:export PYTHONPATH=/home/francis/caffe/python

rm *~ 可考虑删除因修改而自动生成的备份文件 以上可成功配置好cpu_only

安装GPU版本 应先装好CUDA,条件允许可以可加入cudnn,用于加速卷积运算 CUDA安装步骤待后续完善 注释掉CPU-only,其他与前面相同。

关于“Linux下怎么安装卷积神经网络框架caffe”这篇文章的内容就介绍到这里,感谢各位的阅读!相信大家对“Linux下怎么安装卷积神经网络框架caffe”知识都有一定的了解,大家如果还想学习更多知识,欢迎关注编程网行业资讯频道。

阅读原文内容投诉

免责声明:

① 本站未注明“稿件来源”的信息均来自网络整理。其文字、图片和音视频稿件的所属权归原作者所有。本站收集整理出于非商业性的教育和科研之目的,并不意味着本站赞同其观点或证实其内容的真实性。仅作为临时的测试数据,供内部测试之用。本站并未授权任何人以任何方式主动获取本站任何信息。

② 本站未注明“稿件来源”的临时测试数据将在测试完成后最终做删除处理。有问题或投稿请发送至: 邮箱/279061341@qq.com QQ/279061341

软考中级精品资料免费领

  • 历年真题答案解析
  • 备考技巧名师总结
  • 高频考点精准押题
  • 2024年上半年信息系统项目管理师第二批次真题及答案解析(完整版)

    难度     807人已做
    查看
  • 【考后总结】2024年5月26日信息系统项目管理师第2批次考情分析

    难度     351人已做
    查看
  • 【考后总结】2024年5月25日信息系统项目管理师第1批次考情分析

    难度     314人已做
    查看
  • 2024年上半年软考高项第一、二批次真题考点汇总(完整版)

    难度     433人已做
    查看
  • 2024年上半年系统架构设计师考试综合知识真题

    难度     221人已做
    查看

相关文章

发现更多好内容

猜你喜欢

AI推送时光机
位置:首页-资讯-人工智能
咦!没有更多了?去看看其它编程学习网 内容吧
首页课程
资料下载
问答资讯