mmcv-full低版本的安装,与高版本的torch不兼容情况(已经解决)

博客介绍了mmcv-full==1.3.8版本安装技巧。当torch版本过高,用pip或mim安装会报错,原因是高版本torch除去了THC.h文件。解决方案是从github下载源码,在特定文件中注释相关内容、替换代码,再回到源码路径执行安装命令。

摘要生成于 C知道 ,由 DeepSeek-R1 满血版支持, 前往体验 >

mmcv-full==1.3.8版本安装技巧

问题:当torch版本过高的时候,会出现报错:

使用pip或者mim安装时:

# mim install mmcv-full==1.3.8或者
# pip install mmcv-full==1.3.8
报错:
 /tmp/pip-install-2dlg1w3o/mmcv-/tmp/pip-install-2dlg1w3o/mmcv-full_e7921b9ade3a4fe2b0be4d44b13028f9/mmcv/ops/csrc/pytorch/cc_attention_cuda.cu:4:10: fatal error: THC/THC.h: No such file or directory
          4 | #include <THC/THC.h>
            |          ^~~~~~~~~~~
      compilation terminated.

这是由于高版本torch中除去了THC.h这个文件

解决方案:

  1. 在github上下载源码,
  2. 在mmcv/ops/csrc/pytorch/cc_attention_cuda.cu文件中
    • 注释起来 #include <THC/THC.h>
    • 将THCudaCheck替换成AT_CUDA_CHECK(有4处)
  3. 回到源码mmcv的stupy.py路径下,执行:
  4. 执行安装命令(2选1)
MMCV_WITH_OPS=1 pip install -e . #安装mmcv-full版本
pip install -e .  #安装mmcv版本
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包

打赏作者

小兮风

你的鼓励将是我创作的最大动力

¥1 ¥2 ¥4 ¥6 ¥10 ¥20
扫码支付:¥1
获取中
扫码支付

您的余额不足,请更换扫码支付或充值

打赏作者

实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

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

余额充值