Open IUUUUI opened 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。
(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>
环境配置已经完成,但是出现这个问题。请问一下博主。怎么解决呢。