Update TFX Docker image: (#475)
Update TFX Docker image: (#475)

- Install other packages that TFX depends on as well
- Allow per-package version overrides to be specified
- For 'release', always install the latest released version on
  PyPI (rather than a compatible version, which may not be the latest)

Co-authored-by: Chuan Yu Foo <ae8039f1fdad7c527439e91d62a92506c7698445@users.noreply.github.com>
tensorflow/benchmarks,tensorflow/benchmarks