ubuntu16.04 caffe2和detectron安装

最新推荐文章于 2025-05-22 09:25:36 发布

好好学习m

最新推荐文章于 2025-05-22 09:25:36 发布

阅读量270

点赞数

CC 4.0 BY-SA版权

本文链接：https://blog.youkuaiyun.com/weixin_37354297/article/details/88568609

本文详细记录了在Ubuntu环境下安装Caffe2及Detectron的全过程，包括NVIDIA驱动、CUDA、cuDNN的配置，以及解决安装过程中常见问题的方案。同时，提供了检验安装是否成功的步骤。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

虽然只是做毕设需要弄这些东西，但是不知道以后还要安几次，记录一下，以后方便再安装

按照官网流程安装https://caffe2.ai/docs/getting-started.html?platform=ubuntu&configuration=compile

1.安装前需要安装NVIDIA显卡驱动，cuda和cudnn这里可以参考这两篇博客

https://blog.youkuaiyun.com/fdqw_sph/article/details/78745375

https://blog.youkuaiyun.com/qq_38522972/article/details/83504641

安装过程中遇到的问题两篇博客里基本都有提到

安装NVIDIA驱动时如果遇到the nvidia kernel module was not created 的问题，是因为按照教程中给的驱动网址下载的驱动，版本不够新，最好在官网找到符合自己GPU 的驱动下载最新的版本

2.开始安装caffe2（这里因为caffe2已经归入pytorch中了，所以安装caffe2好像也就按张了pytorch)

参考博客：https://blog.youkuaiyun.com/qq_38522972/article/details/83585580

按照官网的步骤走

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 \
      hypothesi

我遇到的问题 pip install这步时遇到了

ImportError: No module named setuptools

解决方法

sudo apt-get install python-setuptools

如果在安装setuptools中遇到 dpkg 被中断，您必须手工运行 ‘sudo dpkg --configur问题

解决方法

sudo rm /var/lib/dpkg/updates/*
sudo apt-get update
sudo apt-get upgrade

复制自己ubuntu版本对应的版本

# for Ubuntu 14.04
sudo apt-get install -y --no-install-recommends \
      libgflags2 \
      cmake3
# for Ubuntu 16.04
sudo apt-get install -y --no-install-recommends \
      libgflags-dev \
      cmake

这一步的问题比较多，等的时间也比较长

git clone https://github.com/pytorch/pytorch.git && cd pytorch
git submodule update --init --recursive
python setup.py install

1.cuda和cudnn版本必须是cuda9.0和对应的cudnn7.0，而不是官网要求的cuda8.0和cudnn6.0，应该是官网没有更新，如果不是这两个版本的话，到第三段代码时候会出错。

2.第二段代码，会出现下载中端的情况，所以在第二段带代码的过程结束时最好再复制回车一下，如果没有出现什么东西再执行第三步

3.第二段代码中，还会出现 missing build dependency:Ubable to 'import.yaml'的报错，在参考的博客中也有提到

解决方法 pip install pyyaml

4.第三段命令中，最好在前面加上sudo，因为我在这一步出现了没有访问权限的报错，加上sudo给他权限

更改环境变量

sudo gedit ~/.bashrc

在文件末尾输入：
export PYTHONPATH=/usr/local:$PYTHONPATH
export PYTHONPATH=$PYTHONPATH:/home/feng/pytorch/caffe2/build  (后面路径为caffe2的编译路径，在caffe2/build中，命令行输入pwd可以得到这个路径)
export LD_LIBRARY_PATH=/usr/local/lib:$LD_LIBRARY_PATH
 
source ~/.bashrc

第二个export中，我没有找到build这个文件夹，自己建了一个，个人觉得应该没问题，毕竟只是一个编译路径，之后应用中再看会不会有问题

检验是否成功

cd ~ && python -c 'from caffe2.python import core' 2>/dev/null && echo "Success" || echo "Failure"

出现success,安装成功

python2 -c 'from caffe2.python import workspace; print(workspace.NumCudaDevices())'

输出大于零就成功了

再在python中

import caffe2

没有报错，成功

安装detectron

参考官网：https://github.com/facebookresearch/Detectron/blob/master/INSTALL.md

官网中是这样的

git clone https://github.com/facebookresearch/detectron $DETECTRON

$DETECTRON 是detectron的下载路径，我不知道这个路径怎么写，直接去掉他，cd到想要保存的地址执行这个命令

git clone https://github.com/facebookresearch/detectron

pip install -r requirements.txt

cd到detectron文件夹执行这个命令

这一步有一个Matplotlib的东西支持python3.4以上的版本安装，将默认python版本改为python3.5（ubuntu安装默认安装好python2.7和python3.5）,用pip3安装

更改为python3.5的方法

sudo update-alternatives --install /usr/bin/python python /usr/bin/python3 150

之后输入

sudo pip3 install -r requirements.txt

如果提示没有安装pip3

sudo apt install python3-pip

更改回python2.7的方法

sudo update-alternatives --config python

在的detectron的位置输入

make

4.测试是否成功

python detectron/tests/test_spatial_narrow_as_op.py

这里需要换成python2.7，因为caffe2是以python2.7安装的。但是不知道用到Matplotlib时会不会出错。

如果出现一个界面，显示ok,和时间就安装成功了