Windows10 + mmdetection(2.13.0) + mmcv(1.3.5) + RTX3070!!!
自己在学mmdetection的时候,查了很多资料发现它对win10很不友好,相应版本又很少,而且相应版本最多适应到cuda10左右,我自己是3070显卡,cuda最少要装到11.0所以苦恼了很久,最后查看mmcv官方文档,发现可以自己编译安装,于是把自己安装过程记录了下来。
mmcv相当于mmdetection的工具包,如果你需要在windows下使用GPU加速你必须要安装mmcv-full,如果你需要使用最新版本的mmdetection或者你的显卡是30系列那么你可以看看接下来教程。
环境配置:
Anaconda3
VS2019一定要安装,下面编译要用到它
CUDA:11.1
Cudnn: 8.04
Python: 3.7
Pytorch: 1.8
还有torchaudio,torchvision什么的,去pytorch.org自己按着pytorch版本安装就行
安装步骤:
(1)安装VS2019这个在网上很多教程,可以自己百度,安装成功后找到以下路径,放到环境变量中,前面可能不同,但是最后肯定是Hostx86\x64
“C:\Program Files (x86)\Microsoft Visual Studio\2019\Community\VC\Tools\MSVC\14.29.30037\bin\Hostx86\x64”
(2)GITHUB上,下载mmcv 和mmdetection相关版本:
MMCV: https://github.com/open-mmlab/mmcv
MMDETECTION:https://github.com/open-mmlab/mmdetection
下载解压过后就如下图这样
(3)打开 poweshell prompt,一定是打开它,不要打开错了!
创建环境:conda create -n mmdet python=3.7
激活环境:conda activate mmdet
(4)安装MMCV(1.3.5)
第三步已经激活了mmdet环境,
首先: cd mmcv-1.3.5文件夹,第三步都在mmcv-1.3.5这个路径下执行的
其次: pip install -r .\requirements.txt 安装相关依赖包,如果这样安装不上,你就打开txt然后自己手动pip install 安装
然后: $env:TORCH_CUDA_ARCH_LIST="8.6” 8.6是我显卡的算力,你们可以根据自己显卡去更改
$env :CV_wITH_Ops = 1 这个不用改
$env:MAX_30BS = 8 这个根据你要动用多少CPU核心去参与MMDET,根据自己CPU核心去算,可以减少到4左右
最后: python setup.py build_ext 报错别怕,是因为我们用的windows中文版的原因直接无视就行
python setup.py develop
(5)安装mmdetection
首先: cd到mmdetection-2.13文件夹下
然后: pip install -r requirements.txt 安装相关依赖包
最后: python setup-py develop 编译一下就行了
到这里就可以啦!这只是我安装过程自己的经验,如果不对的别喷啊!有啥问题留言我们讨论!
禁止转载!!!!