教程很多,但难点在于试合适的环境,试了几次环境终于安装成功了,给我的电脑记一下
环境:CUDA10.2 Python3.8 Ubuntu 18.04
安装要求:
1、conda建环境
conda create -n myconda python=3.8
conda activate myconda
2、安装pytorch(选择CUDA10.2的代码)
conda install pytorch==1.9.0 torchvision==0.10.0 torchaudio==0.9.0 cudatoolkit=10.2 -c pytorch
3、安装opencv、cython、pycocotools
pip install opencv-python
pip install cython
pip install pycocotools
4、安装detectron2
首先git下来
git clone https://github.com/facebookresearch/detectron2.git
然后安装
python -m pip install -e detectron2
之前有过一次报错,应该是GCC版本问题,可选择将GCC降级,降到合适版本,也可以选择直接预安装预安装地址,直接按照环境选择合适的代码粘贴进去安装,但感觉源码安装会更好点