【pytorch1.3+detectron2】安装记录

官方安装就很好 https://github.com/facebookresearch/detectron2/blob/master/INSTALL.md
重要的事情写在前面：
detectron2只支持pytorch1.3及以上， 而pytorch1.3只支持cuda>=9.2， 且pytorch1.5+cuda10.1+detectron2编译时会卡死
0. 安装Cuda+cuDNN（略）
1. anaconda创建虚拟环境：conda create -n pytorch13 python=3.6.8，进入环境，conda activate pytorch13
2. 安装pytorch1.3： conda install pytorch=1.3 torchvision cudatoolkit=10.1 -c pytorch或 pip install torch torchvision推荐后者（安装过了cuda，不必再安装cudatoolkit）
3. 安装opencv： pip install opencv-python==3.4.5.20
4.安装fvcore：pip install 'git+https://github.com/facebookresearch/fvcore'
5. 安装pycocotools:
pip install cython;

pip install 'git+https://github.com/cocodataset/cocoapi.git#subdirectory=PythonAPI'

6. 确保gcc>=4.9 gcc --version
7.安装detectron2

git clone https://github.com/facebookresearch/detectron2.git
cd detectron2
python setup.py build develop

编译的时候可能会报错

Searching for tensorboard
Reading https://pypi.python.org/simple/tensorboard/
No local packages or working download links found for tensorboard
error: Could not find suitable distribution for Requirement.parse('tensorboard')

安装成功
这里安装pip install tensorboard
再次python setup.py build develop通过

采坑
但是，detectron2只支持pytorch1.3， 而pytorch1.3只支持cuda>=9.2,
我的1080ti驱动装的是cuda9.0。。。所以，最后跑程序会报cuda版本不对应的错。。。无法继续跑程序。

填坑
最后，还是妥协了，毕竟显卡驱动在这么底层的东西，靠docker虚拟也是不行的。只好更新显卡驱动到430。见另一帖子【1080TI驱动+CUDA10.1+cudnn】安装记录

最后

以上就是舒适日记本为你收集整理的【pytorch1.3+detectron2】安装记录的全部内容，希望文章能够帮你解决【pytorch1.3+detectron2】安装记录所遇到的程序开发问题。

如果觉得靠谱客网站的内容还不错，欢迎将靠谱客网站推荐给程序员好友。