系统环境:nvidia/cuda:10.1-cudnn7-devel

1. apt-get 安装以下包: libgl1-mesa-glx、cmake、ninja-build、graphviz

apt-get install -y libgl1-mesa-glx cmake ninja-build graphviz

导出成 caffe2 模型时需要 graphviz。

2. conda 安装以下包:

conda install -y tensorboard
conda install -y opencv onnx -c conda-forge
conda install -y pytorch=1.7 torchvision=0.8.1 -c pytorch

导出成 onnx 模型时需要 onnx。

3. 设置环境变量 FORCE_CUDA=”1″

4. 安装 detectron2

pip3 install --user 'git+https://github.com/facebookresearch/fvcore'
git clone https://github.com/facebookresearch/detectron2 detectron2_repo
cd detectron2_repo
pip install --user -e .

  1. https://github.com/facebookresearch/detectron2/tree/master/docker

发表评论

邮箱地址不会被公开。 必填项已用*标注