1.版本选取
- MMDetection的版本选择
安装的时候,需要选择合适的版本,
下载不同版本的MMDetection,从 Github 的历史版本中选择合适的版本
https://github.com/open-mmlab/mmdetection/releases
-
根据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 安装相应的版本
-
mmcv-full的版本
mmcv-full的whl预编译文件地址:
https://download.openmmlab.com/mmcv/dist/index.html
根据cuda、PyTorch的版本选择合适
2.遇到的问题
-
报错
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
-
报错
原因:mmcv-full 版本不匹配
本文内容由网友自发贡献,版权归原作者所有,本站不承担相应法律责任。如您发现有涉嫌抄袭侵权的内容,请联系:hwhale#tublm.com(使用前将#替换为@)