美文网首页
在Ubuntu系统中安装caffe2 2018-11-12

在Ubuntu系统中安装caffe2 2018-11-12

作者: 小核桃mrx | 来源:发表于2018-11-12 22:51 被阅读0次

                                    在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"

        安装成功!

    相关文章

      网友评论

          本文标题:在Ubuntu系统中安装caffe2 2018-11-12

          本文链接:https://www.haomeiwen.com/subject/ayehfqtx.html