How to run Tensorflow on ARM64 devices

Spread the love

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

 

Leave a Reply