Avoid infinite loop in logic to get GPU count

This would be triggered by bad logic in slurm, but we want the test
to be more resilient