Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

windows上编译nms_rotated,总是失败 #317

Closed
liuwenhaha opened this issue May 13, 2022 · 11 comments
Closed

windows上编译nms_rotated,总是失败 #317

liuwenhaha opened this issue May 13, 2022 · 11 comments

Comments

@liuwenhaha
Copy link

环境:windows 10
cuda 11.3
pytorch 1.10
编译一个小时了,总是出问题,有没有大佬能帮我看看
image

@LUO77123
Copy link

按照作者的说明,是可以配好的

@hukaixuan19970627
Copy link
Owner

image

@liuwenhaha
Copy link
Author

image

大佬,我一开始看的就是这个链接,把.cu文件也换了,然后,中文注释都删了,最后还是一样报错

@hukaixuan19970627
Copy link
Owner

hukaixuan19970627 commented May 13, 2022

把以下命令运行一遍,运行结果截个图看看

nvcc -V
nvidia-smi
python
>>> import torch
>>> torch.version.cuda()

@liuwenhaha
Copy link
Author

liuwenhaha commented May 13, 2022

把以下命令运行一遍,运行结果截个图看看

nvcc -V
nvidia-smi
python
>>> import torch
>>> torch.version.cuda()

您帮我看看,感谢!
1、nvcc -V和nvidia-smi
image
2、torch.version.cuda()
image

@hukaixuan19970627
Copy link
Owner

那我无能为力了,不建议windows的原因就是因为每个人都能遇到各式各样的错误,目前只能根据报错提示一点点的改。要是你不急着用的话,七月份之前我更一版不需要用cuda编译的代码

@liuwenhaha
Copy link
Author

那我无能为力了,不建议windows的原因就是因为每个人都能遇到各式各样的错误,目前只能根据报错提示一点点的改。要是你不急着用的话,七月份之前我更一版不需要用cuda编译的代码

谢谢大佬,我不着急。我偶然看到您的代码,因为yolo旋转框目标检测本来就很少。我好像看到detectron2里面也有nms_rotate,我也再研究一下。再次感谢您的回复~

@KolyaIvankov
Copy link

Hey! Sorry, I am no more fluent in Chinese than google translate, but I belive my issue is relevant. Trying to compile on win 11, with CUDA 11.7 and pytorch 1.11 for CUDA 11.3; got message

C:\0\yolov5_obb-master\utils\nms_rotated\src\poly_nms_cuda.cu(4): fatal error C1083: Cannot open include file: 'THC/THC.h': No such file or directory
poly_nms_cuda.cu

googling shows that THC was removed from latest version of pytorch

Could you reproduce this with the newest torch versions?

@hukaixuan19970627
Copy link
Owner

Hey! Sorry, I am no more fluent in Chinese than google translate, but I belive my issue is relevant. Trying to compile on win 11, with CUDA 11.7 and pytorch 1.11 for CUDA 11.3; got message

C:\0\yolov5_obb-master\utils\nms_rotated\src\poly_nms_cuda.cu(4): fatal error C1083: Cannot open include file: 'THC/THC.h': No such file or directory
poly_nms_cuda.cu

googling shows that THC was removed from latest version of pytorch

Could you reproduce this with the newest torch versions?

#152

@hukaixuan19970627 hukaixuan19970627 closed this as not planned Won't fix, can't repro, duplicate, stale May 20, 2022
@lidapengpeng
Copy link

将你的Pytorch版本换成1.10.1,不要用1.11的(这里是为了解决THC.h的问题)

@matteopantano
Copy link

matteopantano commented Jul 19, 2022

These requirements should be added to the install.md @hukaixuan19970627

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

No branches or pull requests

6 participants