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

[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