caffe环境搭建

阿里云2000元红包!本站用户参与享受九折优惠!

caffe需要编译安装,相对tensorflow深度学习框架来说,比较麻烦,但是参照网上教程也安装完毕。下面是步骤以及参考的做法

整体参考的caffe例程是:https://blog.csdn.net/a8039974/article/details/79814531

caffe官网: https://caffe.berkeleyvision.org/installation.html

1, GPU机器 nvidia驱动安装。

请注意驱动版本与GPU机器的硬件型号相关,可在官网查,或者初始购买的机器一般都安装过了

如果自己安装,我这边下载的是NVIDIA-Linux-x86_64-430.26.run。 直接执行sh *.run来进行安装

2, cuda10安装

请注意,cuda版本与nvidia驱动版本有匹配关系,请在网站上查看:

这里用的是

3, cudnn10安装

4, 安装caffe的依赖,以及编译caffe,编译pycaffe

5,  按照例子运行代码

#从本地读取一个视频文件来检测火,安全帽等

cd  /root/work/train-env/py_rfcn

./tools/demo_rfcn.py –net ResNet-50

6,  troubleshooting

1) 运行例子跑测试代码的时候,报错,

error: /io/opencv/modules/highui/src/window.cpp:583: error: (-2) The function is not implemented. Rebuilt the library with Windows, GTK+ 2.x or Carbon support. If you are on Ubuntu or Debian, install libgtk2.0-dev and pkg-config, then re-run cmake or configure script in function cvSowImage

解决办法:

sudo apt-get install pkg-config libgtk2.0-dev pkg-config libavcodec-dev libavformat-dev libswscale-dev libjpeg-dev libpng-dev libtiff-dev libjasper-dev libdc1394-22-dev

然而可能依然没用,接下来又执行pip install opencv-python -i https://pypi.tuna.tsinghua.edu.cn/simple  彻底解决上述问题,重新运行例子,能跑成功

https://www.jianshu.com/p/0b3f2ca86e9c

「点点赞赏,手留余香」

    还没有人赞赏,快来当第一个赞赏的人吧!
0 条回复 A 作者 M 管理员
    所有的伟大,都源于一个勇敢的开始!
欢迎您,新朋友,感谢参与互动!欢迎您 {{author}},您在本站有{{commentsCount}}条评论