一、软硬件信息
服务器厂家: 未知
沐曦GPU型号: C500(4卡,每卡8GB显存)
操作系统内核版本: Linux 5.15.0
是否开启CPU虚拟化: 未知
mx-smi回显: 正常
二、问题现象 我们在沐曦 C500 上配置 BEVFusion 训练环境,遇到环境兼容问题:
需求:
能在沐曦 GPU 上训练 BEVFusion(需要沐曦定制的 mmcv + torch)
也能用 mmdet3d 官方工具生成 nuScenes 数据集的 pkl 文件
问题:
沐曦的 mmcv 2.2.0 是定制版,和标准 mmdet3d 1.x 不兼容
pip 安装 mmdet3d 会检查 mmcv 版本,官方 mmcv ≤1.4.0,但沐曦是 2.2.0
尝试改版本检查绕过后,缺少 mmdet3d.core 模块(新版重构了)
尝试用旧版 mmdet3d (0.11.0),又需要旧版 mmdet,但和沐曦 mmcv 冲突
已生成 pkl 文件,Dataset 可正常加载,但训练报错:
ImportError: cannot import name 'bev_pool_ext'
缺少 bev_pool_ext.cpython-310-x86_64-linux-gnu.so 等 GPU 算子编译文件
请问:
沐曦是否有兼容的 mmdet3d 版本或环境配置方案?
或者有没有 nuScenes 数据集的 pkl 文件可以提供?
BEVFusion 项目需要的 GPU 算子(bev_pool, voxel_layer),是否需要单独编译?
沐曦是否提供这些预编译的 .so 文件?或者有编译脚本可以生成?