MMDetection3D的安装问题,报错缺失libtorch_cuda_cu.so文件

前言

这个系列是记录我个人在学习过程中,遇到的一些问题,并记录我成功的方法,真实有效

MMDetecion3D本身的安装文档非常不利好于新人,所以在此记录了我遇到的坑与解决办法

使用平台

i7-12700k

4070ti

ubuntu20.04

conda虚拟环境中,python=3.8

遇到的问题

根据官方文档操作,前面全部安装成功,包括MMDetction3D也显示安装成功,但是仍然在运行他给出的demo:

python demo/pcd_demo.py demo/data/kitti/000008.bin pointpillars_hv_secfpn_8xb6-160e_kitti-3d-car.py hv_pointpillars_secfpn_6x8_160e_kitti-3d-car_20220331_134606-d42d15ed.pth --show

报错:缺失了libtorch_cuda_cu.so文件

在conda的env路径中搜索,只有libtorch文件而确实没有上述文件。

如下是解决办法

最后发现确实是自己安装的mmcv并不匹配,但是网上的说法都非常的繁琐与笼统,你根本不知道自己究竟应该如何对应,一下是两种我都已经尝试过的成功安装的方法。

方法一:从官网上下载.whl文件

https://download.openmmlab.com/mmcv/dist/cu118/torch2.1.0/index.html

此网址是官方网站,非常重要的一个坑是请你不要直接使用我的网址。我就是因为这样导致的mmcv版本部匹配。

这里有坑

即使是mmcv2.1.0他仍然对应不同的cuda toolkit和pytorch有不同的对应程序,并不代表你下载了mmcv2.1.0就是能保证对应上你自己的cuda和torch的,这是官方文档不会告诉你的东西

网址后缀的cu118和torch2.1.0是可以进行更改的,修改成你的对应版本之后再进行下载。

方法二:命令行直接下载

使用命令

mim install -r requirements/mminstall.tx

这行命令会自动匹配到你对应版本的官网上,我使用之后,对应的就是上述方法一的后缀网站。

然后使用命令

pip install -e.

即可自动安装可使用上述依赖的对应版本mmdet3d,我这里自动安装了1.4.0。

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

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

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

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值