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

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

Open
OliviaNocentini opened this issue Feb 5, 2024 · 3 comments
Open

Comments

@OliviaNocentini
Copy link

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
Copy link
Author

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

@valentin-fngr
Copy link

valentin-fngr commented Mar 12, 2024

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
Copy link

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

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

3 participants