前言
这个系列是记录我个人在学习过程中,遇到的一些问题,并记录我成功的方法,真实有效
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。