Caffe2
ubuntu
电脑
命令如下。（1）sudo apt-get update（2）sudo apt-get install -y --no-install-recommends \ build-essential \ cmake \ git \ libgoogle-glog-dev \ libprotobuf-dev \ protobuf-compiler \ python-dev \ python-pip （3）sudo pip install numpy protobuf
概述。如果您打算使用GPU而不是CPU，那么您应该安装NVIDIA CUDA 8和cuDNN v5.1或v6.0，这是GPU加速的深度神经网络的原生类型库。
首先更新您的显卡驱动程序（NVIDIA CUDA 8）！ 否则您可能会遇到各种难以诊断的错误。1、适用Ubuntu 16.04的命令行。（1）sudo apt-get update && sudo apt-get install wget -y --no-install-recommends（2）wget 'http://developer.download.nvidia.com/compute/cuda/repos/ubuntu1604/x86_64/cuda-repo-ubuntu1604_8.0.61-1_amd64.deb'（3）sudo dpkg -i cuda-repo-ubuntu1604_8.0.61-1_amd64.deb（4）sudo apt-get update（5）sudo apt-get install cuda2、适用Ubuntu 14.04的命令行。（1）sudo apt-get update && sudo apt-get install wget -y --no-install-recommends（2）wget 'http://developer.download.nvidia.com/compute/cuda/repos/ubuntu1404/x86_64/cuda-repo-ubuntu1404_8.0.61-1_amd64.deb'（3）sudo dpkg -i cuda-repo-ubuntu1404_8.0.61-1_amd64.deb（4）sudo apt-get update（5）sudo apt-get install cuda
安装cuDNN（适用所有Ubuntu版本）。（1）CUDNN_URL='http://developer.download.nvidia.com/compute/redist/cudnn/v5.1/cudnn-8.0-linux-x64-v5.1.tgz'（2）wget ${CUDNN_URL}（3）sudo tar -xzf cudnn-8.0-linux-x64-v5.1.tgz -C /usr/local（4）rm cudnn-8.0-linux-x64-v5.1.tgz && sudo ldconfig
libgflags库安装。注意：libgflags2是用于Ubuntu 14.04；libgflags-dev是用于Ubuntu 16.04。这两个不能搞混，否则安装会报错。(1)适用Ubuntu 14.04的命令行。sudo apt-get install -y --no-install-recommends libgflags2(2)适用Ubuntu 16.04的命令行。sudo apt-get install -y --no-install-recommends libgflags-dev
安装其他可选依赖库（适合Ubuntu 14.04和16.04）。（1）sudo apt-get install -y --no-install-recommends \ libgtest-dev \ libiomp-dev \ libleveldb-dev \ liblmdb-dev \ libopencv-dev \ libopenmpi-dev \ libsnappy-dev \ openmpi-bin \ openmpi-doc \ python-pydot（2）sudo pip install \ flask \ future \ graphviz \ hypothesis \ jupyter \ matplotlib \ pydot python-nvd3 \ pyyaml \ requests \ scikit-image \ scipy \ setuptools \ six \ tornado
命令如下。（1）git clone --recursive https://github.com/caffe2/caffe2.git && cd caffe2（2）make && cd build && sudo make install
更改配置文件。（1）打开配置文件的命令行：sudo vim /etc/profile（2）然后在配置文件的最后添加如下代码：export PYTHONPATH=/usr/local:$PYTHONPATHexport PYTHONPATH=$PYTHONPATH:/home/ubuntu/caffe2/buildexport LD_LIBRARY_PATH=/usr/local/lib:$LD_LIBRARY_PATH（3）注意：/home/ubuntu/caffe2/build这部分要根据你的实际目录来填写（export PYTHONPATH=$PYTHONPATH:/home/ubuntu/caffe2/build）。
使修改生效。重新启动使修改配置文件生效。
检查设置是否正确。如果设置正确的话，执行命令以后，会有如下图这样的输出。（1）命令：echo $PYTHONPATH（2）命令：echo $LD_LIBRARY_PATH
caffe2测试。运行下面的命令来测试你的caffe2是否成功。如果输出“Success”，表示成功；如果输出“Failure”，表示失败。（1）python -c 'from caffe2.python import core' 2>/dev/null && echo 'Success' || echo 'Failure'
查看caffe2测试的失败原因。如果上述测试失败了，请输入下面命令 ，查看一下失败的原因。（1）python -c 'from caffe2.python import core'
GPU测试。运行下面的命令来测试你的GPU安装是否成功。（1）python -m caffe2.python.operator_test.relu_op_test