美文网首页
Caffe 安装参考

Caffe 安装参考

作者: NoneLand | 来源:发表于2017-02-17 11:52 被阅读303次

安装参考资料

source /etc/profile 立即生效

ubuntu系统已崩。。。。。就在安装Caffe的依赖库之后,改变源。。。然后就Kernel panic, 再也没法启动了。。。。。
Linux系统make出错之后再次make会在之前的基础上继续,所以需要删除make出错的东西重新make。具体到caffe就是caffe-master下的.build_release文件夹。(直接用make clean
caffe的配置过程[Trouble-shooter]
sudo cp ~/anaconda/lib/libhdf5* /usr/lib/x86_64-linux-gnu/ libhdf5_hl.so.10无法链接问题,坑巨大,真是服了,将anaconda里的文件复制到一个文件夹里面去

make pycaffe 之后只需将路径添加进caffe所在路径添加到PYTHONPATH中,如:
export PYTHONPATH=/path/to/caffe/python:$PYTHONPAT
[] 如Downloads文件夹在/home/noneland下面,而不是在/home下面,就是因为这个原因导致python无法导入caffe库。也是醉了。

其实只要一步一步来,最后确实也能安装成功。大不了吐点血。。。在这个过程中确实学到了很多东西。。。比如OpenCV的编译安装[cv2.so在release/lib下面]。。好心酸。。。不过终于安装成功了。。。

测试MNIST数据集

  • Ubuntu安装Caffe并训练MNIST
    不得不说,Linux下面的确方便,训练这个模型只需要几个命令就搞定了。下载速度也很快。。成功运行截图如下:
训练完成截图

GPU和CPU训练对比

条件: MNIST数据集:50000训练,10000测试;迭代10000次
耗时如下:
CPU(i7-3630QM):14min39sec
CPU(i7-7700K):9min51s
GPU(GT 645M CUDA8.0 CUDNN5.0): 2min43sec
GPU(GTX 1070 CUDA8.0 CUDNN5.1):18sec

从以上对比明显可以看出GPU对于深度学习的重要性了,加速太明显了。

相关文章

网友评论

      本文标题:Caffe 安装参考

      本文链接:https://www.haomeiwen.com/subject/cknkwttx.html