记录安装mmcv-full==1.3.0过程中的问题

记录安装mmcv-full==1.3.0过程中的问题

原因

项目需要测试Swin Transformer for segmentation的效果，发现网上好多都是利用mmcv库，比如，

Instance Segmentation任务在paper with code网站上的官方代码上，

SwinTransformer/Swin-Transformer-Semantic-Segmentation: This is an official implementation for “Swin Transformer: Hierarchical Vision Transformer using Shifted Windows” on Semantic Segmentation. (github.com)

于是乎，这边就需要安装mmcv这个包。官方给的code中mmcv的版本必须>=1.1.4, <=1.3.0，然后官方默认命令安装的版本是1.7.0，不匹配，代码运行就报错，因此就要在本地安装mmcv==1.3.0的版本。

环境

• 系统：win11

• 显卡：Nvidia RTX 3060

安装过程遇到的问题：

首先需要安装VS2019的C++编译环境，（mmcv要求）这个不做多解释，直接安装visual studio2019以及其默认的桌面C++环境即可。

然后说一说，这个过程主要有两个坑，

1，torch版本不能太高，因为其对应的CUDA版本太高会导致mmcv-full库无法安装，因为其一些依赖在最新的torch库中已经被移除了，比如说“THC/THC.h”，参考：

关于THC/THC.h: No such file or directory解决办法_无名°的博客-CSDN博客

2，torch版本不能太低，因为RTX 30系显卡不兼容低版本的CUDA运行，目前个人测试，1.5版本（不可行），1.8版本（可行），参考：

Install mmcv, mmdetection on RTX30 series (velog.io)

总结：

1，安装pytorch1.8

# conda
conda install pytorch==1.8.1 torchvision==0.9.1 torchaudio==0.8.1 -c pytorch

官方文档：Previous PyTorch Versions | PyTorch

2，安装mmcv-full==1.3.0

pip install -U openmim
mim install mmcv-full==1.3.0

官方文档：Installation — mmcv 1.7.0 documentation

2022.11.8 更新

好吧，pytorch1.8好像也不支持30系显卡，上面方法好像行不通，当作一次记录吧。