安装后训练报错，请问怎么解决

[RadioWaverider]

/gemini/code/yolov5_obb Traceback (most recent call last): File "export.py", line 57, in from models.common import Conv File "/gemini/code/yolov5_obb/models/common.py", line 23, in from utils.datasets import exif_transpose, letterbox File "/gemini/code/yolov5_obb/utils/datasets.py", line 28, in from utils.augmentations import Albumentations, augment_hsv, copy_paste, letterbox, mixup, random_perspective File "/gemini/code/yolov5_obb/utils/augmentations.py", line 12, in from utils.general import LOGGER, check_version, colorstr, resample_segments, segment2box File "/gemini/code/yolov5_obb/utils/general.py", line 35, in from utils.nms_rotated import obb_nms File "/gemini/code/yolov5_obb/utils/nms_rotated/init.py", line 1, in from .nms_rotated_wrapper import obb_nms, poly_nms File "/gemini/code/yolov5_obb/utils/nms_rotated/nms_rotated_wrapper.py", line 4, in from . import nms_rotated_ext ImportError: /gemini/code/yolov5_obb/utils/nms_rotated/nms_rotated_ext.cpython-38-x86_64-linux-gnu.so: undefined symbol: _ZNK2at10TensorBase8data_ptrIdEEPT_v

训练代码： !python train.py
--weights 'weights/yolov5n.pt'
--data 'data/yolov5obb_demo.yaml'
--hyp 'data/hyps/obb/hyp.finetune_dota.yaml'
--epochs 10
--batch-size 1
--img 1024
--device 0