I solved the iterator issue by calling the following before the program exits:
data_iter._shutdown_workers()
I solved the display issue by calling the following immediately after plt.imshow:
plt.show()