I'd start with very short distances, and pick a toy the pup gets really excited about.
A leash helps too, so once the pup picks up the toy you can guide them back toward you.
And of course, lots of praise and rewards!
The pup needs to learn that coming to you (either with a toy or ball, or without) is the greatest thing in the world!
Good luck!!