安装mmocr
一、安装mmdetection
在安装前,如果已经安装过mmcv,先卸载掉,否则不同版本会导致ModuleNotFoundError报错!
1、先安装对应版本的pytorch(本次cuda10.2,pytorch1.7)
2、安装对应版本的mmcv
3、安装request.txt文件中的依赖库,以上工作完毕后,在安装mmdection,以下为具体详细步骤
1、安装miniconda,创建虚拟环境
conda create -n open-mmlab python=3.7 -y
conda activate open-mmlab
2、安装对应版本的pytorch(本次cuda10.2,所以安装pytorch1.7)
因为mmdetection依赖pytorch,所以版本必须对应好
pytorch官网:https://pytorch.org/get-started/previous-versions/
# CUDA 10.2 -i 加的百度镜像,提提速
pip install torch==1.5.0 torchvision==0.6.0 -i https://mirror.baidu.com/pypi/simple
3、安装对应版本的mmcv-full
此处需要注意安装的版本问题:
pip install mmcv-full -f https://download.openmmlab.com/mmcv/dist/{cu_version}/{torch_version}/index.html
需要将上述语句url中 的**{cu_version}** 和 {torch_version} 替换成你对应的cuda版本和安装的pytorch版本
例如:我安装的cuda为10.2,pytorch为1.7,则执行语句如下所示:
pip install mmcv-full -f https://download.openmmlab.com/mmcv/dist/cu102/torch1.7.0/index.html
执行完毕如下:
4、安装MMDetection库
# 现在源码项目
git clone https://github.com/open-mmlab/mmdetection.git
cd mmdetection
# 安装对应的依赖库和MMDetection
pip install -r requirements/build.txt
pip install -v -e . # or "python setup.py develop"
安装完毕:
5、安装mmocr
# install mmocr
cd {Path to TableMASTER_mmocr}
pip install -v -e .