记录安装mmcv-full==1.3.0过程中的问题
原因
项目需要测试Swin Transformer for segmentation的效果,发现网上好多都是利用mmcv库,比如,
Instance Segmentation任务在paper with code网站上的官方代码上,
于是乎,这边就需要安装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系显卡,上面方法好像行不通,当作一次记录吧。