一、annaconda安装虚拟环境
conda create -n detectron python=2.7
二、安装caffe2 Caffe2官网安装教程
conda install pytorch-nightly cudatoolkit=9.0 -c pytorch
注意,对于cuda版本为9.2的,如果直接使用conda install pytorch-nightly -c pytorch命令,则默认安装的是10.0的版本,会导致后面出错"CUDA driver version is insufficient for CUDA runtime version"。
三、安装future、protobuf
pip install matplotlib==2.1 protobuf future pycocotools Cython
四、对照官网安装