在Ubuntu系统中安装caffe2
安装顺序:
安装cuda--------安装cudnn---------源码编译安装Caffe2
安装cuda的教程我的上一篇文章已经详细说明了,接下来就写剩下的两个步骤。
1.安装cudnn
1.1 去官网下载cudnn安装包
1.2解压缩cuDNN包。
sudo tar -xzvf cudnn-9.0-linux-x64-v7.tgz
1.3将以下文件复制到CUDA Toolkit目录中,然后更改文件权限。
sudo cp cuda/include/cudnn.h /usr/local/cuda/include
sudo cp cuda/lib64/libcudnn* /usr/local/cuda/lib64
sudo chmod a+r /usr/local/cuda/include/cudnn.h /usr/local/cuda/lib64/libcudnn*
2.安装caffe2
2.1安装依赖项
sudo apt-get update
sudo apt-get install-y--no-install-recommends\
build-essential\
git\
libgoogle-glog-dev\libgtest-dev\
libiomp-dev\libleveldb-dev\
liblmdb-dev\libopencv-dev\
libopenmpi-dev\
libsnappy-dev\
libprotobuf-dev\
openmpi-bin\
openmpi-doc\
protobuf-compiler\
python-dev\
python-pip
pip install--user\
future\
numpy\
protobuf\
typing\
hypothesis
如果出现以下错误:
它的解决方法是:
2.2 for Ubuntu 16.04
sudo apt-get install-y--no-install-recommends\
libgflags-dev\
cmake
2.3克隆与构建
git clone https://github.com/pytorch/pytorch.git && cd pytorch
git submodule update--init-- recursive
sudo python setup.py install
3.测试Caffe2安装
cd ~ && python -c 'from caffe2.python importcore' 2>/dev/null && echo "Success" || echo "Failure"
安装成功!
网友评论