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。

你可能感兴趣的:(BEVformer,python,ubuntu,计算机视觉,目标检测,机器学习,神经网络,深度学习)