Ubuntu caffe 搭建

最新推荐文章于 2024-05-11 23:32:26 发布

原创最新推荐文章于 2024-05-11 23:32:26 发布 · 861 阅读

1 ·

本内容遵循CC 4.0 BY-SA版权协议

ubuntu 同时被 2 个专栏收录

4 篇文章

订阅专栏

深度学习

2 篇文章

订阅专栏

本文详细介绍如何在Ubuntu系统上搭建Caffe深度学习框架。主要包括NVIDIA驱动和CUDA的安装、依赖库安装、Atlas和cuDNN配置以及Caffe编译步骤。特别针对GPU加速、依赖管理及编译过程中可能出现的问题提供了指导。

Ubuntu caffe搭建

1、安装NVIDIA驱动和CUDA
2、安装caffe依赖
sudo apt-get install libprotobuf-dev libleveldb-dev libsnappy-dev libopencv-dev libboost-all-dev libhdf5-serial-dev libgflags-dev libgoogle-glog-dev liblmdb-dev protobuf-compiler python-pip python-dev python-opencv
3、安装Atlas
sudo apt-get install libatlas-base-dev
4、安装cudnn
https://developer.nvidia.com/cudnn
解压
tar -xzvf cudnn-**
cd cuda/lib64
sudo cp * /usr/local/cuda/lib64
cd ../include
sudo cp cudnn.h /usr/local/cuda/include
更新软链接
/usr/local/cuda/lib64

5、安装caffe
a)进入caffe根目录，首先复制一份 Makefile.config , 然后修改里面的内容，主要需要修改的参数包括
USE_CUDNN 是是否使用cudnn加速
CPU_ONLY 是否只使用CPU模式，没有GPU没安装CUDA的同学可以打开这个选项
BLAS (使用intel mkl还是atlas)
MATLAB_DIR 如果需要使用MATLAB wrapper的同学需要指定matlab的安装路径, 如我的路径为 /usr/local/MATLAB/R2013b (注意该目录下需要包含bin文件夹，bin文件夹里应该包含mex二进制程序)
DEBUG 是否使用debug模式，打开此选项则可以在eclipse或者NSight中debug程序
b)编译
make –j8 all
make -j8 test
make runtest
注意 -j8 是指使用几个线程来同时编译，可以加快速度， j后面的数字可以根据CPU core的个数来决定。
c)CMake 生成cmake程序的所有引用
在CAFFE_ROOT目录下
mkdir cmakebuild
cd cmakebuild
cmake ..
make all -j8
make runtest
d)以下步骤可选择：
编译Matlab wrapper
执行如下命令
make matcaffe
然后就可以跑官方的matlab demo啦。
编译Python wrapper
make pycaffe