A Tensorflow container with Intel and ARM support

7 December 2021

I was kind of surprised about discovering that the official TensorFlow docker image on Docker Hub does not support the ARM architecture, and thus the M1 Apple silicon chips, or at least as of today. So I built one: https://hub.docker.com/r/sarusso/tensorflow.

The sarusso/tensorflow:v2.7.0 image available at the above link supports both Intel (amd64) and ARM/M1 (arm64) architectures. Only the v2.7.0 tag exists, so it is mandatory to set it if using this image.