专栏算法工具链maptrv2复现报错

maptrv2复现报错

已解决
CSY2024-07-01
98
1

在4090上复现maptr报错

File "/home/guxunjia/project/MapTR_v2/projects/mmdet3d_plugin/maptr/modules/encoder.py", line 1110, in forward

x, depth = super().forward(images, img_metas)

File "/home/guxunjia/anaconda3/envs/maptr_v2/lib/python3.8/site-packages/mmcv/runner/fp16_utils.py", line 214, in new_func

output = old_func(*new_args, **new_kwargs)

File "/home/guxunjia/project/MapTR_v2/projects/mmdet3d_plugin/maptr/modules/encoder.py", line 282, in forward

geom = self.get_geometry_v1(

File "/home/guxunjia/anaconda3/envs/maptr_v2/lib/python3.8/site-packages/mmcv/runner/fp16_utils.py", line 214, in new_func

output = old_func(*new_args, **new_kwargs)

File "/home/guxunjia/project/MapTR_v2/projects/mmdet3d_plugin/maptr/modules/encoder.py", line 115, in get_geometry_v1

torch.inverse(post_rots)

RuntimeError: CUDA error: operation not supported when calling cusparseCreate(handle)

算法工具链
评论1
0/1000
  • 潘俊伟
    Lv.1

    可以更新pytorch环境或者使用下面示例对代码中相应部分进行修改

    torch.inverse(lidar2ego_rots.to("cpu")).to("cuda:0")

    2024-07-01
    0
    0