一、mmcv与mmcv-full的区别:
关于mmcv-full和mmcv的区别官网上是这样写的:总结一下就是需要CPU和CUDA算子就安装mmcv-full版本
关于mmcv-full的官网介绍及安装链接放在这里:
二、安装时的注意事项-pytorch等版本问题
mmcv库的版本与pytorch,CUDA的版本要对应,对于mmcv>=2.0以上的版本,直接根据mmcv官网的指令安装即可
但是对于mmcv1.x的版本没有具体说明,我尝试过使用CUDA11.6+pytorch11.3的组合,失败了
也尝试过在自己的电脑上编译源码,在wheel的时候都遇到了同样的报错
[debug] fatal error: THC/THC.h: No such file or directory
在官网上看了一圈,发现官网从源码编译mmcv-full的介绍里使用要求有如下依赖
所以最终尝试了CUDA10.2及对应pytorch然后去安装mmcv-full
三、安装方法
根据官网建议
pip install -U openmim
mim install mmcv-full==1.2.7
最后安装成功
PS.在其他版本cuda和pytorch下安装会有各种各样的报错,最终原因估计都是pytorch版本问题