pytorch/vision

windows maskrcnn error: unresolved external symbol "class at::Tensor __cdecl nms_cpu(class at::Tensor const &,class at::Tensor const &,float)" (?nms_cpu@@YA?AVTensor@at@@AEBV12@0M@Z)

Open

#2,139 创建于 2020年4月24日

在 GitHub 查看
 (22 评论) (0 反应) (0 负责人)Python (15,050 star) (6,858 fork)batch import
help wantedmodule: c++ frontendmodule: opswindows

描述

I built torchvision pulled from master using :

cmake -DCMAKE_PREFIX_PATH=D:/Garage/onnx-deploy/libtorchd  -DCMAKE_INSTALL_PREFIX=D:/vision/Debug -DCMAKE_BUILD_TYPE=Debug ..
cmake --build . --config Debug 
cmake --install .

Then torch.jit.script(model) and saved mask-rcnn model from torchvision. Errors occured in my C++ project, firstly Unknown builtin op: torchvision::_new_empty_tensor_op, then I modified my code according to #1407, added lines like #include "torchvision/nms.h" and torch::RegisterOperators().op("torchvision::nms", &nms) , but it won't work. The error became unresolved external symbol "class at::Tensor __cdecl nms_cpu(class at::Tensor const &,class at::Tensor const &,float)" (?nms_cpu@@YA?AVTensor@at@@AEBV12@0M@Z). The libtorch version is 1.5.0 and torchvision is pulled from master yesterday. Any clue how I can get mask rcnn running? Similar issues are many but little to help, I really get confused about how to modify CMakeList.txt of TorchVision.

cc @peterjc123 @nbcsm @guyang3532 @maxluk @gunandrose4u @smartcat2010 @mszhanyi

贡献者指南

技术栈
cpppython
领域
build systembackend
议题类型
bug
难度面向新贡献者的预计实现难度，1 表示很小改动，5 表示专家级工作。
4
预计时间有经验贡献者完成调查、实现、测试并准备 pull request 的粗略时间范围。
over 1 week
活动状态议题当前的可参与程度：新鲜、活跃、陈旧、阻塞或等待维护者输入。
stale
清晰度议题是否清楚说明期望改动、验收标准和下一步。
needs investigation
前置要求
CMake basicsWindows C++ compilationtorchvision custom opslibtorch setup
新手友好度1-100 的估计分数，表示该议题对首次贡献者的友好程度。
10
研究方向
The issue involves a linking error for nms cpu on Windows. Look at torchvision's CMakeLists.txt to verify the NMS op is built and linked correctly. Check the comments in the issue for possible workarounds (e.g., #1407 mentions operator registration). Investigate the Windows build configuration (e.g., use of CMAKE PREFIX PATH, Debug vs Release) and ensure the torchvision library exports nms cpu symbol. Reference the file 'torchvision/csrc/ops/nms.h' and test with a minimal C++ project.