Ubuntu18.04和windows10配置实例分割模型SOLOV2

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不行,而安装过程默认最新
安装步骤

这里不给出具体的命令,只说安装步骤:

  1. 自己百度查看怎么安装显卡驱动,然后安装好cuda,网上很多教程
  2. 安装pip命令,再安装好pytroch,注意版本
  3. 安装mmcv,安装参考官方https://github.com/open-mmlab/mmcv/tree/v0.2.16
  4. 安装mmdection,安装命令参考官方https://github.com/open-mmlab/mmdetection
  5. 安装SOLOV2,安装命令参考windows中的链接
  6. 这里面需要修改一些代码,防止出现安装过程中默认安装最新的库的情况,这样会导致安装失败,主要修改的地方在mmcv文件夹中,找到opencv-python>=某个版本这个地方,改为4.1.1.26,mmdet和solo中也需要在requirements文件夹中的所有txt文件添加上opencv-python4.1.1.26,solo中可以在requirements文件夹中将torch版本相关删除,防止重新安装,前提是torch你已经安装好了。
  7. 后续会补上百度云链接,有制作好的安装包

可能你会觉得我好像没说啥,可是真要自己装,版本的问题真的要一点点去踩,安装成功就点个赞

你可能感兴趣的:(pytorch)