Caffe2 是一个兼具表现力、速度和模块性的深度学习框架,是 Caffe 的实验性重构,能以更灵活的方式组织计算。由 FaceBook 开源,该框架可以用在 iOS、Android 和树莓派上训练和部署模型。
建立 Caffe2
详细的构建矩阵:
git clone --recursive https://github.com/caffe2/caffe2.git cd caffe2
OS X
brew install automake protobuf mkdir build && cd build cmake .. make
Ubuntu
可运行版本:
Ubuntu 14.04
Ubuntu 16.06
需要的依赖包
sudo apt-get update sudo apt-get install -y --no-install-recommends \ build-essential \ cmake \ git \ libgoogle-glog-dev \ libprotobuf-dev \ protobuf-compiler \ python-dev \ python-pip sudo pip install numpy protobuf
可选择 GPU 支持
如果你计划使用 GPU,而不只是使用 CPU,那你应该安装 NVIDIA CUDA 和 cuDNN,这是一个面向深度神经网络的 GPU 加速库。英伟达在官方博客中详细介绍了安装指南,或者可以尝试下面的快速安装指令。首先,一定要升级你的图显驱动!否则你可能遭受错误诊断的极大困难。
构建环境(已知能运行)
评论