美文网首页
在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