主要参考了官方文档Welcome to MMDetection3D’s documentation! — MMDetection3D 0.17.3 文档
和知乎答主文章
(一)mmdetection3d 环境配置与KITTI 数据集使用 - Jose Cruise的文章 - 知乎 https://zhuanlan.zhihu.com/p/427879646
最后成功的配置是cuda11.1+torch1.8.0+mmcv1.3.8+mmdet2.14.0+pycocotools2.0.1+mmseg0.14.1 和 mmdet3d0.17。
安装完成后,可能会有numpy报错,这个时候的numpy版本是1.19.5,在网上查到有些方法是升级numpy版本,但是不可以,会导致后续报错。
这个时候把pycocotools降级到2.0.1版本可以解决问题。
同时发现,numba0.48.0不支持cuda11.3,但是在cuda11.1上可以跑通。
ubuntu系统下切换cuda版本的方法是用runfile安装不同版本的cuda和cudnn,随后通过建立不同的软连接来切换,同时修改bashrc,此处参考了linux 创建连接命令 ln -s 软链接 - 小 楼 一 夜 听 春 雨 - 博客园