Detectron2,基于 Pytorch 。
记录一下Detectron2 环境搭建过程,弄了一下午,踩了好多坑。
我们先来看看官方的配置需求
Requirements
Linux or macOS with Python ≥ 3.6
PyTorch ≥ 1.4
torchvision that matches the PyTorch installation. You can install them together at pytorch.org to make sure of this.
pycocotools. Install it by pip install pycocotools>=2.0.1.
OpenCV, optional, needed by demo and visualization
环境配置
本人系统centos7(无GPU)、anaconda3
anaconda虚拟环境
conda create -n py37 python=3.7
conda env list
source activate py37
注意:后面所有的安装指令都在py37虚拟环境中运行
pytorch环境
进入官网选择自己对应的版本
注意:去掉后面的 -c pytorch
根据该命令的提示很容易就能安装好 Pytorch。
conda install pytorch torchvision cpuonly
如果安装失败,很可能是因为网络的延迟过高而超时。
所以我们可以通过配置 anacoda 国内镜像源来加速安装。
命令执行以下命令
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main/