Ubuntu16.04+CUDA8.0+caffe配置

来源:互联网 发布:中国三大网络巨头 编辑:程序博客网 时间:2024/06/08 13:11

    最近由于工作需求,需要搭建caffe框架。

    收藏了一篇可供参考的博客, 但博客中有一些需要改动的地方,在这里根据自己遇到的坑进行一下纠正。

1.安装依赖项改为

$ sudo apt-get install -y libprotobuf-dev libleveldb-dev libsnappy-dev libopencv-dev libboost-all-dev libhdf5-serial-dev libgflags-dev libgoogle-glog-dev liblmdb-dev protobuf-compiler protobuf-c-compiler protobuf-compiler python-numpy python-scipy python-matplotlib python-sklearn python-skimage python-h5py python-protobuf python-leveldb python-networkx python-nose python-pandas python-gflags cython ipython    原文中的依赖项不够,导致最开始编译caffe的时候,出现很多莫名其妙的错误。2.配置cudnn的时候需要将文件按如下进行拷贝
$ sudo cp include/cudnn.h /usr/local/include$ sudo cp lib64/libcudnn.* /usr/local/lib$ sudo ln -sf /usr/local/lib/libcudnn.so.5.1.3 /usr/local/lib/libcudnn.so.5$ sudo ln -sf /usr/local/lib/libcudnn.so.5 /usr/local/lib/libcudnn.so$ sudo ldconfig -v
如果不这样做,在编译caffe的时候会报出lcudnn找不到的错误。

其他配置与原文相同即可,

原创粉丝点击