ROS2 не может импортировать имя «iou3d_nms_cuda» ⇐ Python

[Anonymous]

Я новичок в использовании ROS в своем проекте. Я установил ROS2 на Ubuntu 22.04. Я успешно протестировал свой код глубокого обучения Python (не узел ROS) для обнаружения объектов Pointcloud с помощью anaconda. Когда я использую print(torch.version.cuda) в Spyder, я получаю 11,7.

Я создал подписчика Python ROS2 для глубокого обучения обнаружению трехмерных объектов. Я использовал sys.path.insert(0,r'/home/user/MachinLearning/'), чтобы добавить каталог всех кодов/модулей глубокого обучения на свой узел подписчика.

Для версии cuda: когда я использую nvcc --version, я получаю следующее:

nvcc: драйвер компилятора NVIDIA (R) Cuda Авторские права (c) 2005–2021 гг., корпорация NVIDIA Построено: четверг_18_ноября_09:45:30_PST_2021. Инструменты компиляции Cuda, выпуск 11.5, V11.5.119 Сборка cuda_11.5.r11.5/compiler.30672275_0 Для узла подписчика: когда я использую ros2 run object_detection прослушиватель, я получаю эту ошибку (iou3d_nms_cuda находится в папке MachinLearning):

из . импортировать iou3d_nms_cuda Ошибка ImportError: невозможно импортировать имя «iou3d_nms_cuda» из «ops.iou3d_nms» (неизвестное местоположение) [ros2run]: процесс завершен с ошибкой 1 Как я могу решить эту проблему?

Заранее благодарен за любую помощь.

Спасибо

Аббас