error when run the setup.py

[2111905222]

detector/nms/src/nms_cuda.cpp:4:23: error: ‘AT_CHECK’ was not declared in this scope; did you mean ‘DCHECK’?
4 | #define CHECK_CUDA(x) AT_CHECK(x.type().is_cuda(), #x, " must be a CUDAtensor ")
| ^~~~~~~~
detector/nms/src/nms_cuda.cpp:9:3: note: in expansion of macro ‘CHECK_CUDA’
9 | CHECK_CUDA(dets);
| ^~~~~~~~~~
error: command 'gcc' failed with exit status 1

this error occur no matter conda or pip. who can help me pls!! should I reinstall cuda?

[Fang-Haoshu]

what is the pytorch version you are using?

[Fang-Haoshu]

if version>=1.5, please merge this PR: #592

[coldmanck]

Just a note: do the merge it by this commandgit pull origin pull/592/head

[iniesta123]

Hi ,

I was not able to install alphapose on my system by following the instructions mentioned in this repo.

I have tried with different versions of python, torch, torch-vision, gcc and cuda but no luck [ on ubuntu18.04] ...

This is the error I am getting when I am using versions of ( pytorch==1.6 torchvision==0.7 cudatoolkit=10.2 python=3.6 gcc=8.4 )
" FAILED: /home/messi/Alphapose/AlphaPose/build/temp.linux-x86_64-3.6/detector/nms/src/nms_kernel.o "

It would be great if someone can suggest solution for the same

Thanks in advance !!