error: command ‘C:\\Program Files\\NVIDIA GPU Computing Toolkit\\CUDA\\v10.2\\bin\\nvcc.exe‘ failed

无码不欢的我

已于 2022-02-28 23:17:27 修改

阅读量1.7w

点赞数 29

分类专栏： bug信息解决方案文章标签：目标检测深度学习人工智能

于 2022-02-28 22:59:28 首次发布

本文链接：https://blog.csdn.net/m0_56192771/article/details/123194532

版权

bug信息解决方案专栏收录该内容

10 篇文章 0 订阅

订阅专栏

我在windows10下安装detectron2（最新版maskRCNN）时，遇到了错误信息：detectron2 ImportError: cannot import name ‘_C‘ from ‘detectron2
解决办法：
切换到 setup.py 文件所在目录，然后在终端命令行中输入命令：python setup.py build develop

如果出现如下错误：
error: command ‘C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.2\bin\nvcc.exe’ failed with exit status 1

解决办法：
找到如下文件
detectron2\layers\csrc\nms_rotated\nms_rotated_cuda.cudetectron2\layers\csrc\nms_rotated\nms_rotated_cuda.cu
修改一下部分内容（将部分代码注释掉）：
其中原文件的内容如下所示：

// Copyright (c) Facebook, Inc. and its affiliates.
#include <ATen/ATen.h>
#include <ATen/cuda/CUDAContext.h>
#include <c10/cuda/CUDAGuard.h>
#include <ATen/cuda/CUDAApplyUtils.cuh>
#ifdef WITH_CUDA
#include "../box_iou_rotated/box_iou_rotated_utils.h"
#endif
// TODO avoid this when pytorch supports "same directory" hipification
#ifdef WITH_HIP
#include "box_iou_rotated/box_iou_rotated_utils.h"
#endif

需要修改为：

// Copyright (c) Facebook, Inc. and its affiliates.
#include <ATen/ATen.h>
#include <ATen/cuda/CUDAContext.h>
#include <c10/cuda/CUDAGuard.h>
#include <ATen/cuda/CUDAApplyUtils.cuh>
/*#ifdef WITH_CUDA
#include "../box_iou_rotated/box_iou_rotated_utils.h"
#endif
// TODO avoid this when pytorch supports "same directory" hipification
#ifdef WITH_HIP
#include "box_iou_rotated/box_iou_rotated_utils.h"
#endif*/
#include "box_iou_rotated/box_iou_rotated_utils.h"

然后再次在终端命令行中输入命令：python setup.py build develop
就可以了。