Unfortunately, Tensorflow doesn’t have docker builds for arm64, only for amd64. But you can use third-party build from another vendors.

I found builds from www.arm.com

• https://github.com/ARM-software
• https://github.com/ARM-software/Tool-Solutions/tree/main/docker/tensorflow-aarch64
• https://hub.docker.com/u/armswdev
• https://hub.docker.com/r/armswdev/tensorflow-arm-neoverse/tags

So you can just pull Tensorflow image for arm64:

docker pull armswdev/tensorflow-arm-neoverse:latest

and run with mount option (-v)

docker run -it -v /your_path/:/your_path_in_docker armswdev/tensorflow-arm-neoverse:latest bash