I wanted to run a pose estimation algorithm using Tensorflow on Nvidia’s Jetson TX2 platform. (https://github.com/eldar/pose-tensorflow)
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 it was expected 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?