Base images for running tenserflow scripts