Egrt / yolov7-obb

在YOLOv7的基础上使用KLD损失修改为旋转目标检测yolov7-obb
GNU General Public License v3.0
176 stars 26 forks source link

ImportError: DLL load failed while importing nms_rotated_ext: 找不到指定的程序。 #36

Open IUUUUI opened 11 months ago

IUUUUI commented 11 months ago

环境配置已经完成,但是出现这个问题。请问一下博主。怎么解决呢。

IUUUUI commented 11 months ago

博主,您好。每次训练到10个epoch时,出现 inds = nms_rotated_ext.nms_rotated(dets_th, scores, iou_thr) RuntimeError: Not compiled with GPU support 的错误 下载nms_rotated时,出现No CUDA runtime is found, using CUDA_HOME='D:\cuda\NVIDIA GPU Computing Toolkit\CUDA\v11.3' Compiling nms_rotated_ext without CUDA running build_ext D:\Anaconda3\lib\site-packages\torch\utils\cpp_extension.py:476: UserWarning: Attempted to use ninja as the BuildExtension backend but we could not find ninja.. Falling back to using the slow distutils backend. warnings.warn(msg.format('we could not find ninja.'))的错误, 我的cuda是11.3,pytorch是1.12,python是3.8。

IUUUUI commented 11 months ago

(pytorch) D:\1yolov5_rotate\yolov7_obb\yolov7_obb\utils\nms_rotated>python Python 3.8.5 (default, Sep 3 2020, 21:29:08) [MSC v.1916 64 bit (AMD64)] :: Anaconda, Inc. on win32 Type "help", "copyright", "credits" or "license" for more information.

import torch torch.version '1.12.1' nvcc-V Traceback (most recent call last): File "", line 1, in NameError: name 'nvcc' is not defined exit()

(pytorch) D:\1yolov5_rotate\yolov7_obb\yolov7_obb\utils\nms_rotated>nvcc -V nvcc: NVIDIA (R) Cuda compiler driver Copyright (c) 2005-2021 NVIDIA Corporation Built on Sun_Mar_21_19:24:09_Pacific_Daylight_Time_2021 Cuda compilation tools, release 11.3, V11.3.58 Build cuda_11.3.r11.3/compiler.29745058_0

(pytorch) D:\1yolov5_rotate\yolov7_obb\yolov7_obb\utils\nms_rotated>