您的位置：首页 > Web前端

Ubuntu16.04安装caffe2(含detectron配置)

2020-01-14 18:46 1226 查看

Ubuntu16.04安装caffe2和detectron安装配置

安装显卡驱动
CUDA安装
安装anaconda
Caffe2安装
安装detectron

在介绍流程前，先放上我所参考的一些博客，并且十分感谢这些博主，感谢他们之前的尝试，为我安装caffe2提供便利，文章里有些图是来源于这些博客的，以下为我所参考的博客：
https://blog.csdn.net/wanzhen4330/article/details/81706405
https://www.geek-share.com/detail/2723529179.html
https://www.geek-share.com/detail/2710760484.html
https://blog.csdn.net/qq_29462849/article/details/85841388

我这个安装的过程之前只有重装了ubuntu16.04这一操作，没有其他多余操作，为了方便，我把ubuntu16.04安装过程也放上来，大家可以参考下面的链接https://cloud.tencent.com/developer/news/270017
我是按照这个链接一步步来安装ubuntu16.04的。
在安装caffe2前需要安装cuda和显卡驱动，在安装前做一些检查。
在终端输入lspci | grep -i nvidia可以检查gpu的型号，看是否支持cuda
输入uname -m && cat /etc/*release ，用来检查自己的Linux版本是否支持 CUDA，结果如下，证明是可以的

在终端输入gcc -version ，若出现

则表示已安装，否则输入sudo apt-get install build-essential 进行安装

在终端输入uname -r 结果为

在终端输入sudo apt-get install linux-headers-$(uname -r) ，可以安装对应kernel版本的kernel header和package development，结果显示为

在完成以上步骤后，可以开始安装显卡驱动和cuda

安装显卡驱动

要先安装显卡驱动再安装cuda,安装显卡驱动前需先禁用ubuntu自带的驱动nouveau，通过下面的操作可以禁用nouveau
在终端输入
sudo vi /etc/modprobe.d/blacklist-nouveau.conf

就会进入到一个文件
在文件里添加下面两条语句
(需要注意的是，这里编辑用到的是ubuntu自带的vi编辑软件，使用方法需要自己百度，我自己用的时候就是在输入完上面的语句以后，按一下a，然后把下面的两条语句自己打上去，在打完下面的语句以后按下ESC，然后输入!wq，再按一下回车)
blacklist nouveau
options nouveau modeset=0

在按下!wq保存退出编辑后，继续输入
sudo update-initramfs -u

然后重启电脑
重启电脑后，会发现分辨率改变了，这时候不用着急，在ubuntu右上角找到系统设置，进入后找到“软件和更新”，选择“附加驱动”，如下图所示

然后点击右下角的应用更改，等待一下，看到重启电脑选项，点击重启。重启后会发现分辨率变正常了。这样nvida显卡驱动就安装好了。

CUDA安装

参照博客的第一种方法
我装的是cuda9.0
https://blog.csdn.net/wanzhen4330/article/details/81699769#cudnn的安装
主要步骤：
1、在存放cuda下载包的文件夹打开终端，并输入语句sudo sh cuda_9.0.176_384.81_linux.run ，其中”sh”后的是你所下载的cuda下载包的名字

2、在开始安装后，会出现很长的服务条款，一直按enter键直至其显示完，然后分别按照问题输入accept n(不安装driver) y y y，其中会遇到关于defult安装位置的问题(这类问题没有选项)，直接按enter键就可以,然后等待其安装。

3、安装完成后要设置环境变量，打开Home目录下的 .bashrc文件添加如下路径，如果没有找到，则在Home目录下下按Ctrl+H键显示隐藏文件，找到.bashrc文件后，在…bashrc 文件(在用户目录下)的尾部写入
export PATH=/usr/local/cuda-9.1/bin:$PATH

export LD_LIBRARY_PATH=/usr/local/cuda-9.1/lib64:$LD_LIBRARY_PATH
我在.bashrc文件所在的目录下打开终端也输入上面的两条语句

4、在完成上面步骤后在终端运行一下 source ~/.bashrc (这个我是直接在桌面打开的终端运行的)
在终端输入nvcc --version 看是否安装成功，若出现

只要最后一句出现了你所下载的cuda版本号就可以了

但是如果出现了图里最后一句的是版本号为7.5，则利用下面这个博客解决
https://blog.csdn.net/weixin_43229347/article/details/85229783

即先输入sudo apt-get autoremove nvidia-cuda-toolkit
再添加环境变量export PATH=$PATH:/usr/local/cuda/bin
最后再查看一次版本号，就发现cuda版本号是自己想要的版本号

安装anaconda

https://www.geek-share.com/detail/2710760484.html
主要步骤：
1.去anaconda官网下载anaconda安装包
2.打开存放anaconda下载包的文件夹，在这个文件夹打开终端，输入
bash Anaconda2-4.4.0-Linux-x86_64.sh
然后像安装cuda一样，按enter键读完条款，不过这次不用选no，一直yes和enter键
安装结束后，关掉终端，重新打开一个新的终端，输入python，若看见

python号为自己安装的版本号就成功了，这个图介绍了python的版本信息，但是后面带了anaconda的标识，这就说明Anaconda安装成功了，此时输入exit()退出python环境，输入conda list，会看到一大堆packages,说明成功了，如果你在这一步输入python之后显示的还是系统默认的版本，那说明.bashrc的更新还没有生效，命令行输入：source ~/.bashrc，就可以成功安装。

Caffe2安装

直接用官方文档
https://caffe2.ai/docs/getting-started.html?platform=windows&configuration=compile

选择环境

安装依赖包
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
对应下图

选择对应的系统
sudo apt-get install -y --no-install-recommends
libgflags-dev
Cmake

对应于下图

下图的步骤是用来安装cuda的，因为之前的步骤已经安装好了，所以这一步我们可以跳过

来到这一步就是开始安装caffe2
git clone https://github.com/pytorch/pytorch.git && cd pytorch
git submodule update --init --recursive
python setup.py install
对应于下图

安装过程需要一些时间