cannot import name 'roi_align_cuda' from 'alphapose.utils.roi_align

MVIG-SJTU / AlphaPose

Real-Time and Accurate Full-Body Multi-Person Pose Estimation&Tracking System

http://mvig.org/research/alphapose.html

Other

7.83k stars 1.96k forks source link

cannot import name 'roi_align_cuda' from 'alphapose.utils.roi_align #1199

Open OliviaNocentini opened 5 months ago

OliviaNocentini commented 5 months ago

Hi, I installed Alphapose following the guide steps. When I run the demo for human pose tracking, I'm getting the following error while running the file:

./scripts/inference.sh --pose_track

prova

I have installed python3.7 and my nvidia version is 11.5

Can anybody help me? I tried also this command suggested in the issue #1061 but it is not working for me conda install pytorch==1.12.1 torchvision==0.13.1 torchaudio==0.12.1 cudatoolkit=11.3 -c pytorch

Thank you!

OliviaNocentini commented 4 months ago

I solved the problem, I had to comment the line from . import roi_align_cuda in roy_align.py

valentin-fngr commented 3 months ago

As an additional comment : when using CPU only, you might want to comment this line : from . import nms_cpu, nms_cuda

You can always add an if statement to check if any cuda visible devices are found, and if so, import the gpu utils

qy-0320-888 commented 3 months ago

你好，我也遇到了同样的问题，请问怎么解决了？