debug: NameError: name ‘_C‘ is not defined 本地运行 GroundingDINO 代码 debug 记录-Toy模板网

这篇具有很好参考价值的文章主要介绍了debug: NameError: name ‘_C‘ is not defined 本地运行 GroundingDINO 代码 debug 记录。希望对大家有所帮助。如果存在错误或未考虑完全的地方，请大家不吝赐教，您也可以点击"举报违法"按钮提交疑问。

在本地跑 GroundingDINO 代码（github）

首先down下来代码：
git clone https://github.com/IDEA-Research/GroundingDINO.git
然后跟着 readme 走，先下载预训练参数放到 ./weight 文件夹：
mkdir weights
cd weights
wget -q https://github.com/IDEA-Research/GroundingDINO/releases/download/v0.1.0-alpha/groundingdino_swint_ogc.pth

然后新开一个 py 文件，复制 readme 上的代码运行：

from groundingdino.util.inference import load_model, load_image, predict, annotate
import cv2

model = load_model("groundingdino/config/GroundingDINO_SwinT_OGC.py", "weights/groundingdino_swint_ogc.pth")
IMAGE_PATH = "weights/dog-3.jpeg"
TEXT_PROMPT = "chair . person . dog ."
BOX_TRESHOLD = 0.35
TEXT_TRESHOLD = 0.25

image_source, image = load_image(IMAGE_PATH)

boxes, logits, phrases = predict(
    model=model,
    image=image,
    caption=TEXT_PROMPT,
    box_threshold=BOX_TRESHOLD,
    text_threshold=TEXT_TRESHOLD
)

annotated_frame = annotate(image_source=image_source, boxes=boxes, logits=logits, phrases=phrases)
cv2.imwrite("annotated_image.jpg", annotated_frame)

然后报错和 debug，这个错误都是普适性错误，第二次犯了，所以记录下来，防止再犯

NameError: name ‘_C’ is not defined

省流

运行
pip3 install -q -e .

详细过程

之前跑另一个模型的时候也遇到这个问题，勉强记得是因为没有编译模型。down 下来的代码有个 setup.py 文件，于是我尝试运行：
python setup.py

但是报错：error: no commands supplied

然后发现后面应该加上 build 或者 install，所以我重新运行：
python setup.py install
python setup.py build
但是还是不行，还是报错

查到大家在讨论这个问题：NameError: name ‘_C’ is not defined

说应该设置环境变量 CUDA_HOME，原因是，查 setup 代码：

   if CUDA_HOME is not None and (torch.cuda.is_available() or "TORCH_CUDA_ARCH_LIST" in os.environ):
       print("Compiling with CUDA")
       extension = CUDAExtension
       sources += source_cuda
       define_macros += [("WITH_CUDA", None)]
       extra_compile_args["nvcc"] = [
           "-DCUDA_HAS_FP16=1",
           "-D__CUDA_NO_HALF_OPERATORS__",
           "-D__CUDA_NO_HALF_CONVERSIONS__",
           "-D__CUDA_NO_HALF2_OPERATORS__",
        ]