fix: Mac M1 compatible Dockerfile and bump TF version

[terrytangyuan]

The TF image does not work on Mac M1. Switching to use Python base image and install TF from there. Also bumped TF versions.

Verified that the example works.

[google-oss-prow]

[APPROVALNOTIFIER] This PR is APPROVED

This pull-request has been approved by: terrytangyuan

The full list of commands accepted by this bot can be found here.

The pull request process is described here

Needs approval from an approver in each of these files:

• OWNERS [terrytangyuan]

Approvers can indicate their approval by writing /approve in a comment
Approvers can cancel approval by writing /approve cancel in a comment

[coveralls]

Pull Request Test Coverage Report for Build 3706744342

• 0 of 0 changed or added relevant lines in 0 files are covered.
• 2 unchanged lines in 1 file lost coverage.
• Overall coverage decreased (-0.1%) to 39.455%

---

Files with Coverage Reduction	New Missed Lines	%
pkg/controller.v1/mpi/mpijob_controller.go	2	77.67%

Totals
Change from base Build 3655920262:	-0.1%
Covered Lines:	2681
Relevant Lines:	6795

---

💛 - Coveralls

[johnugeorge]

/lgtm