mmcv-full==1.2.7+pytorch+cuda配置问题

本文讲述了mmcv与mmcv-full的区别,重点在于mmcv-full需要额外的CPU和CUDA支持。安装过程中需要注意版本匹配,尤其是PyTorch和CUDA的版本,作者推荐CUDA10.2和对应PyTorch版本,并分享了遇到的问题和解决方法。
摘要由CSDN通过智能技术生成

一、mmcv与mmcv-full的区别:

关于mmcv-full和mmcv的区别官网上是这样写的:总结一下就是需要CPU和CUDA算子就安装mmcv-full版本

关于mmcv-full的官网介绍及安装链接放在这里:

安装 MMCV — mmcv 1.7.2 文档

二、安装时的注意事项-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版本问题

  • 7
    点赞
  • 9
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值