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

support pytorch1.11 #152

Open
wants to merge 1 commit into
base: master
Choose a base branch
from

Conversation

QingYuan-L
Copy link

@QingYuan-L QingYuan-L commented Jan 11, 2022

@hukaixuan19970627
Copy link
Owner

/home/test/Persons/hukaixuan/yolov5/utils/nms_rotated/src/poly_nms_cuda.cu:5:27: fatal error: ATen/ceil_div.h: No such file or directory
compilation terminated.
ninja: build stopped: subcommand failed.

It seems ’ATen/ceil_div.h‘ is missing in Torch1.10.0.

@QingYuan-L
Copy link
Author

we can wait for pytoch1.11(stable) to test this pr, if you use the newest NGC image, maybe the pr work. my dev environment : cuda11.5 pytorch1.11 ,nvcr.io/nvidia/pytorch:21.12-py3 from NGC

@Prabhakaran-Gokul
Copy link

Prabhakaran-Gokul commented Jun 22, 2022

Thanks for the PR @QingYuan-L. In addition to the changes that you have made, I had to comment out THC/THC.h and include c10/cuda/CUDACachingAllocator.h in the poly_nms_cuda.cu file to make it compile. For example:

//#include <THC/THC.h>
#include <c10/cuda/CUDACachingAllocator.h>

CUDA version: 11.7
Nvcc version: 11.7
PyTorch version: 1.11.0
PyTorch cuda version (ie. torch.version.cuda): 11.3

@lcw99
Copy link

lcw99 commented Jun 27, 2022

Same environment, but I need to change THCudaCheck to C10_CUDA_CHECK to compile.

@Ni-Wo
Copy link

Ni-Wo commented Jul 11, 2022

I'm running into the same problem, I'm quite new to this field and don't understand everything here. Could someone help me out with a more detailed explanation what to do?
My environment:

  • Windows10
  • Python 3.9
  • nvcc 11.3, nvidia-smi: 11.6
  • torch: 1.12
    I already commented out the "#include <THC/THC.h>" and added "#include <c10/cuda/CUDACachingAllocator.h>". But I didn't understand the comments before that.

@LUO77123
Copy link

目前支持pytorch1.11吗

@tinly00
Copy link

tinly00 commented Jul 19, 2022

have you solved this problem?

@LUO77123
Copy link

I'm running into the same problem, I'm quite new to this field and don't understand everything here. Could someone help me out with a more detailed explanation what to do? My environment:

  • Windows10
  • Python 3.9
  • nvcc 11.3, nvidia-smi: 11.6
  • torch: 1.12
    I already commented out the "#include <THC/THC.h>" and added "#include <c10/cuda/CUDACachingAllocator.h>". But I didn't understand the comments before that.

have you solved this problem?

@shotyme
Copy link

shotyme commented Jan 10, 2023

My solution to compile successfully was to downgrade torch and change the requirements.txt to:

torch==1.9.1
torchvision==0.10.1

@LUO77123
Copy link

LUO77123 commented Jan 10, 2023 via email

@myknowntime
Copy link

你好,请问方便提供一下相关代码或者指点一下迷津吗?

@yjoh25
Copy link

yjoh25 commented Oct 18, 2023

hi?
Has the above error been resolved?

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

Successfully merging this pull request may close these issues.

10 participants