Demo codes not working on TX2 GPU

[karthee320]

Hi,

I wanted to run your pose estimation algorithm on Nvidia's Jetson TX2 platform.

While running in CPU by explicitly specifying CUDA_VISIBLE_DEVICES=-1, both the single person and multi-person pose estimation worked and gave proper results as you shown in the repository and the results are consistent .

But when I try to run the same code on GPU(CUDA_VISIBLE_DEVICES=0), the code works without any errors but the results are wrong. I tried to print the output (outputs_np['part_prob']), it is giving me different results every time I run.

On side note, when I try to run on my desktop GPU (GeForce GTX 1050 Ti), the results are correct and consistent.

Is there anything I can do to correct this behavior?