目录

一、前言

二、安装python,CUDA, Pytorch

2.1版本对应

2.2python安装

2.3CUDA安装

2.4pytorch安装

三、安装mmcv-full,mmdet

四、安装编译版本mmcv

五、编译版mmcv2.0.1,mmdet3.1.0 


一、前言

mmcv>=2.0.0的版本不再区分mmcv与mmcv-full

包名<2.0>=2.0
mmcv-full包含CUDA算子
mmcv不包含CUDA算子包含CUDA算子
mmcv-lite不包含CUDA算子

详细区别参考:https://zhuanlan.zhihu.com/p/572550592

需要注意的是,mmcv-full只能在pytorch的1.x.0版本上进行编译,所以再确定需要版本为1.x.x时,要查看自己的pytorch版本,若不符合需要重新创建新环境。尤其注意CUDA版本

对于从github上下载的项目,首先查看mmdet文件下的__init__.py和version.py文件,确定该项目用的mmcv版本和mmdet版本。

我这里指定mmcv版本在1.3.17至1.7.0之间的版本,mmdet为2.25.3版本

接下来是我的安装历程,我安装的版本信息如下

pythonCUDApytorchmmcv-fullmmdet
3.710.21.8.01.7.02.25.3

二、安装python,CUDA, Pytorch

2.1版本对应

下图为python与torch,torchvision版本对应,来源于Torch 、torchvision 、Python 版本对应关系以及安装 GPU 或 CPU 版本的 pytorch_torch torchvision_xzw96的博客-CSDN博客

下图为cudatoolkit与torch版本对应

cudatoolkittorch1.5torch1.6torch1.7torch1.8torch1.9torch1.10torch1.11
9.2
10.1
10.2
11.0
11.1
11.3

2.2python安装

链接:https://pan.baidu.com/s/1o_jfQq-P-WtCWRZOJ0Dn8g?pwd=zll8 
提取码:zll8

网盘连接放这里,其中有Anaconda的安装文件以及方法 ,这个自行安装即可

2.3CUDA安装

参考连接:http://t.csdnimg.cn/KiK3V

试了很多博主的方法,该博主讲的最细以及方法最适用

2.4pytorch安装

首先打开anaconda prompt,创建一个新的虚拟环境,指定python版本,如我的3.7

conda create -n mcvful python=3.7

由于我是在安装前更换了电脑CUDA的版本,也不知道哪里没有处理好,所以在创建新的环境时出现了CondaVerificationError错误,如下:

CondaVerificationError: The package for wheel located at /home/xxx/anaconda3/pkgs/wheel-0.36.2-pyhd3eb1b0_0
appears to be corrupted. The path 'site-packages/wheel/vendored/__init__.py'
specified in the package manifest cannot be found.

CondaVerificationError: The package for wheel located at /home/xxx/anaconda3/pkgs/wheel-0.36.2-pyhd3eb1b0_0
appears to be corrupted. The path 'site-packages/wheel/vendored/packaging/__init__.py'
specified in the package manifest cannot be found.

 如果出现类似错误,请参考:http://t.csdn.cn/w6xAc

我的是该文件出现错误,按照该博主的步骤,删除后再重新创建环境就没问题了

环境创建成功后激活环境

activate mcvful

 然后下载torch1.8.0和torchvision0.9.0(根据自己torch版本找对应的torchvision版本)

链接:download.pytorch.org/whl/torch_stable.html

下载好后分别复制torch和torchvision文件的路径

分别运行下面段代码

pip install "E:\ana\envs\mcvful\torch-1.8.0-cp39-cp39-win_amd64.whl"

pip insatll "E:\ana\envs\mcvful\torchvision-0.9.0-cp39-cp39-win_amd64.whl"

如果报如下的错误

Could not find a version that satisfies the requirement pymysql (from versions: none)

可以在后面加一个pip源,如:

pip install "E:\ana\envs\mcvful\torch-1.8.0-cp39-cp39-win_amd64.whl" -i https://pypi.tuna.tsinghua.edu.cn/simple

torchvision同理

安装完成后进行pip list查看

测试torch安装是否成功,true则表示成功

三、安装mmcv-full,mmdet

下图中打  的位置表示存在该版本的mmcv-full包,CUDA10.2的兼容性最好(github官网说的),所以尽量安装CUDA10.2,

CUDAtorch1.11torch1.10torch1.9torch1.8torch1.7torch1.6torch1.5
11.5
11.3
11.1
11.0
10.2
10.1
9.2
cpu

官网给出了基本的安装步骤:开始你的第一步 — MMDetection 3.1.0 文档,如果完全按照官网指示安装,容易安装成最新版本的,我的步骤如下:

按照官网的要求运行前两句,

pip install -U openmim
mim install mmengine

由于我需要安装mmcv-full-1.7.0,所以运行如下语句(最好加一个清华源,不然容易下载失败,mmdet同理):

mim install mmcv-full==1.7.0 -i https://pypi.tuna.tsinghua.edu.cn/simple

 然后安装mmdet,直接指定版本mmdet==2.25.3(需要注意的是如果是下载的别人的基于mmcv的项目,最好不要用清华源安装mmdet,尤其是需要在该项目上进行改动时,尽量安装编译版的mmcv,用setup.py去安装,原因参照http://t.csdnimg.cn/herQd)

mim install mmdet==2.25.3 -i https://pypi.tuna.tsinghua.edu.cn/simple

四、安装编译版本mmcv

参考Windows 环境从零安装 mmcv-full - 知乎 (zhihu.com)

其中编译时打开的是

而不是 

 打开后先通过cd到源码下载目录,然后激活环境如

 这里激活必须使用conda命令,然后遵照作者所说编译即可,需要注意的是运行顺序:

  1. 安装开发依赖项(如果有):pip install -r requirements-dev.txt
  2. 运行 setup.pypython setup.py develop
  3. 安装运行时依赖项:pip install -r requirements.txt

五、编译版mmcv2.0.1,mmdet3.1.0 

参考从源码编译 MMCV — mmcv 2.0.1 文档

Logo

旨在为数千万中国开发者提供一个无缝且高效的云端环境,以支持学习、使用和贡献开源项目。

更多推荐