MMDetection= Cuda10.0.130 + Pytorch1.4.0 + torchvision0.5.0 + mmcv-full1.1.6 + mmdet2.5.0

1.版本选取

  1. MMDetection的版本选择
    安装的时候,需要选择合适的版本,
    下载不同版本的MMDetection,从 Github 的历史版本中选择合适的版本
    https://github.com/open-mmlab/mmdetection/releases

MMDetection= Cuda10.0.130 + Pytorch1.4.0 + torchvision0.5.0 + mmcv-full1.1.6 + mmdet2.5.0_第1张图片

  1. 根据Cuda的版本,cudatoolkit=10.0.130,选择PyTorch 和 torchvision 的版本
    Pytorch 、torchvision 的预编译whl文件下载地址,
    https://download.pytorch.org/whl/torch_stable.html
    可以直接下载whl文件,然后

    pip install xxx.whl
    

    或者

    pip install torch==1.4.0+cu100 -f https://download.pytorch.org/whl/torch_stable.html
    

    如果

    conda install pytorch torchvision 
    

    安装的torch、torchvision版本不对应,可以使用 pip 安装相应的版本

  2. mmcv-full的版本
    mmcv-full的whl预编译文件地址:
    https://download.openmmlab.com/mmcv/dist/index.html
    根据cuda、PyTorch的版本选择合适

2.遇到的问题

  1. 报错

    ImportError: libtorch_cpu.so: cannot open shared object file: No such file or directory
    

    PyTorch=1.4.0时,mmcv-full=1.3.0时,报这个错;
                  mmcv-full=1.2.1时,不报错;
    原因:mmcv-full 版本太高
    测试mmcv-full是否安装成功

    from mmcv.ops import RoIPool
    
  2. 报错
    在这里插入图片描述
    原因:mmcv-full 版本不匹配

你可能感兴趣的:(mmdetection)