caffe2安装记录

caffe安装记录

依赖项安装

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

源码下载及安装

git clone --recursive https://github.com/caffe2/caffe2.git && cd caffe2
make && cd build && sudo make install

注意：若在github上用zip格式下载，可能会出现若干第三方包(caffe2/third_party/)下载不下来的错误，因此请使用git clone，即上面的代码进行下载和安装；

测试安装是否成功：

python -c 'from caffe2.python import core' 2>/dev/null && echo "Success" || echo "Failure"
输出success则安装成功

GPU测试：

python -m caffe2.python.operator_test.relu_op_test

成功：
<caffe2.proto.caffe2_pb2.DeviceOption at 0x7ff19b6237d0>], engine=u'CUDNN')
.
----------------------------------------------------------------------
Ran 1 test in 4.858s

OK

错误：

Cudnn.h: No such file or directory

报错：

In file included from
/home/hw/main-data/00_softwares/caffe2/caffe2/core/context_gpu.cu:10:0:
/home/hw/main-data/00_softwares/caffe2/caffe2/core/common_cudnn.h:7:19: fatal error: cudnn.h: No such file or directory

原因：cudnn未安装

解决方案：下载：https://developer.nvidia.com/rdp/cudnn-download

cuDNN v7.0 Library for Linux
#解压文件
tar -zxvf cudnn-6.5-linux-x64-v2.tgz
#切换路径
cd cudnn-6.5-linux-x64-v2
#复制lib文件到cuda安装路径下的lib64/
sudo cp lib* /usr/local/cuda/lib64/
#复制头文件
sudo cp cudnn.h /usr/local/cuda/include/

建议：直接切换到su管理员权限，进行复制文件操作；网上说的需要更新软链接，我装的时候没有用上，补充在下边：

更新软连接

cd /usr/local/cuda/lib64/
sudo rm -rf libcudnn.so libcudnn.so.6.5
sudo ln -s libcudnn.so.6.5.48 libcudnn.so.6.5
sudo ln -s libcudnn.so.6.5 libcudnn.so

错误：

No module named past.builtins

解决方案：

Sudo pip install future

安装不成功，请更换pip源：

sudo vim ~/.pip/pip.conf

添加：

[global]
index-url = http://mirrors.aliyun.com/pypi/simple/ [install]
trusted-host=mirrors.aliyun.com

错误：

No module named hypothesis

解决方案：

sudo pip install hypothesis

错误：在build文件夹下运行示例成功，其他文件夹错误

环境变量未设置好，python运行时不知道搜索路径，请做如下操作

echo $PYTHONPATH
export PYTHONPATH=/usr/local:$PYTHONPATH
export PYTHONPATH=$PYTHONPATH:/home/hw/main-data/00_softwares/caffe2/build
echo $LD_LIBRARY_PATH
export LD_LIBRARY_PATH=/usr/local/lib:$LD_LIBRARY_PATH

如果还是不行：

vim / etc/bash.bashrc
末尾添加
export PYTHONPATH=/usr/local:$PYTHONPATH
export PYTHONPATH=$PYTHONPATH:/home/hw/main-data/00_softwares/caffe2/build
export LD_LIBRARY_PATH=/usr/local/lib:$LD_LIBRARY_PATH