-
Notifications
You must be signed in to change notification settings - Fork 431
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
Comments
按照作者的说明,是可以配好的 |
把以下命令运行一遍,运行结果截个图看看 nvcc -V
nvidia-smi
python
>>> import torch
>>> torch.version.cuda() |
那我无能为力了,不建议windows的原因就是因为每个人都能遇到各式各样的错误,目前只能根据报错提示一点点的改。要是你不急着用的话,七月份之前我更一版不需要用cuda编译的代码 |
谢谢大佬,我不着急。我偶然看到您的代码,因为yolo旋转框目标检测本来就很少。我好像看到detectron2里面也有nms_rotate,我也再研究一下。再次感谢您的回复~ |
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
googling shows that THC was removed from latest version of pytorch Could you reproduce this with the newest torch versions? |
|
将你的Pytorch版本换成1.10.1,不要用1.11的(这里是为了解决THC.h的问题) |
These requirements should be added to the install.md @hukaixuan19970627 |
环境:windows 10
cuda 11.3
pytorch 1.10
编译一个小时了,总是出问题,有没有大佬能帮我看看
The text was updated successfully, but these errors were encountered: