CUDA error when running demo.sh/imgs2poses.py

[hangg7]

Hi authors, thanks for open-sourcing the code of your amazing paper! I've been encountering with a few issues when trying to run through your demo.

I have followed your README to install the provided nvidia-docker. But it seems that when running demo.sh inside of the docker, CUDA errors exist. More specifically, when running this line, program complaints about "PyramidCU::GenerateFeatureList: an illegal memory access was encountered". I've prepared my running log after both with and without cuda-memcheck for debugging purpose. Could you please help me identify where could possibly go wrong?

I'm using RTX 2080 Ti + cuda-10.1 + NVIDIA driver 418.39 on Ubuntu 18.04. Not sure if being too updated could be the problem.

[rodrygojose]

Hi @hangg7 , do you have more than one GPU on your machine?
Could you please set the environment variable CUDA_VISIBLE_DEVICES.

[hangg7]

Hi @hangg7 , do you have more than one GPU on your machine?
Could you please set the environment variable CUDA_VISIBLE_DEVICES.

Sweet!

CUDA_VISIBLE_DEVICES=0 bash demo.sh

perfectly fixed my issue. Maybe it's also worth updating the README as well? Since no such instructions are currently listed.

Closing now, and thanks for such a quick response!