这几天需要在window下配置detectron2,差点搞得我精神崩溃,差不多弄了有3,4天,网上的方法差不多都试了试,结果还是各种报错。本来想在淘宝上找大佬帮装,结果一问价格要80大洋(༼ಢ_ಢ༽)学生党遭不住。就在昨天凌晨一点,我终于配置成功,并成功跑出一个小demo,仅以此篇记录一下我的安装过程,相当于笔记,方便下次的安装,也希望可以帮助一些小伙伴。
安装VS2019——MSVC编译环境
具体的下载安装流程大家可以自行搜索,这里提供参考:VS2019安装和使用教程(超详细)_vs2019安装教程_小敏敏呀的博客-CSDN博客
尽量使用VS2019,VS2022也行但不推荐,如果你是VS2022,可以参考这位博主修改:
【安装教程】Windows10/11安装detectron2教程_detectron2安装_K.K.U的博客-CSDN博客
这是旧版本的下载地址:
Visual Studio 较旧的下载 - 2019、2017、2015 和以前的版本
安装conda
参考这位:Anaconda超详细安装教程(Windows环境下)_conda安装_菜鸟1号!!的博客-CSDN博客
安装cuda+cudnn
参考这位:
CUDA和cudnn的安装_MT_Lucky的博客-CSDN博客
这里需要注意
1.自己的显卡和CUDA的匹配
2.detectron2支持的cuda版本(如下图)
3.cudnn和CUDA的匹配。
像我是30系列的显卡(RTX 3050Ti),使用cuda11.x版本的。我下载的是cuda11.1,这里我也推荐下载cuda11.1
![](https://img-blog.csdnimg.cn/dff0f1f9c7484253a83542d9de12b23c.png)
创建并配置环境
参考这位:
Detectron2安装踩坑记录(比较详细版)_光芒再现robotliang的博客-CSDN博客
这位博主写的十分详尽,大家可以参考。
但我安装的是deteron2-windows版,所以从目录3.1完成后就不再参考博主,大家也可以试试跟着博主安装,里面对于一些报错的解决方案还是比较详细的。
同时提醒一下几点
1.环境python的版本设为3.7为好。
2.这里的pytorch版本最好安装1.9.0的,因为我安装detectron-windows版本时,pytorch1.10.0环境下一直报错,但更换1.9.0后可以正常安装。
3.注意pytorch和CUDA版本的匹配,先前的pytorch版本在这:Previous PyTorch Versions | PyTorch
如果你是CUDA11.1且想下载pytorch1.9.0,可以直接使用以下指令安装
pip install torch==1.9.0+cu111 torchvision==0.10.0+cu111 torchaudio==0.9.0 -f https://download.pytorch.org/whl/torch_stable.html
在安装完成之后,需要检测CUDA和torch版本是否匹配参考这位博主:
检查cuda和pytorch版本是否匹配_little TangTang的博客-CSDN博客
安装
pycocotools-windows和detectron2-windows
参考这位博主:(从章节三开始执行)
Windows下安装detectron2(免修改版本)_detectron2 windows_微雨曳荷的博客-CSDN博客
执行完操作即可安装成功:
附 pip list:
执行demo测试
demo的提供者:detectron2安装详细教程+demo测试_LiYikou的博客-CSDN博客
结果:
这就是我的全部安装过程啦!
这里面我肯定有很多表达不足之处,请各位大佬多多指点!
希望和大家一起进步!!!