windows配置
环境需求
- cuda 9.2 硬性要求,否则失败
- torch 1.2.0
- torchvision 0.4.0
- C ++ 2015及以上
- mmdetection 1.1.0
- mmcv 新的可以用
- SOLOV2 新的可以用
- opencv-python 新的可以用
安装步骤
主要的安装步骤可以参考B站上的UP主- 小鸡炖技术 ,他出了很多环境安装的视频,我之前安装就是参考了他的视频,但是环境要求一定要参考我的!
这里是链接:https://www.bilibili.com/video/BV1M5411s784/?spm_id_from=333.788.videocard.6
这个是安装的mmdetection的视频,安装好这个SOLO就很简单了,按照官方代码中给出的安装命令就好,链接:https://github.com/WXinlong/SOLO/blob/master/docs/INSTALL.md
Ubuntu配置
环境需求
- cuda 10.1 别11.1就行
- torch 1.4.0 硬性要求
- torchvison 0.5.0
- gcc 7.5.0 这个版本我装了系统就有,版本依赖性不太大
- opencv-python 4.1.1.26 最新的4.4.0不行,而安装过程默认最新
安装步骤
这里不给出具体的命令,只说安装步骤:
- 自己百度查看怎么安装显卡驱动,然后安装好cuda,网上很多教程
- 安装pip命令,再安装好pytroch,注意版本
- 安装mmcv,安装参考官方https://github.com/open-mmlab/mmcv/tree/v0.2.16
- 安装mmdection,安装命令参考官方https://github.com/open-mmlab/mmdetection
- 安装SOLOV2,安装命令参考windows中的链接
- 这里面需要修改一些代码,防止出现安装过程中默认安装最新的库的情况,这样会导致安装失败,主要修改的地方在mmcv文件夹中,找到opencv-python>=某个版本这个地方,改为4.1.1.26,mmdet和solo中也需要在requirements文件夹中的所有txt文件添加上opencv-python4.1.1.26,solo中可以在requirements文件夹中将torch版本相关删除,防止重新安装,前提是torch你已经安装好了。
- 后续会补上百度云链接,有制作好的安装包
可能你会觉得我好像没说啥,可是真要自己装,版本的问题真的要一点点去踩,安装成功就点个赞