armswdev/tensorflow-arm-neoverse-n1

By armswdev

Updated about 3 years ago

Please use "armswdev/tensorflow-arm-neoverse"

Image
Data Science
Internet of Things
Machine Learning & AI

5.1K

TensorFlow docker images for Arm Neoverse-N1

WARNING: This repository is no longer being updated. Please see armswdev/tensorflow-arm-neoverse for newer containers.

More information, and a Getting Started guide are available from ARM-Software's Tool-Solutions repository on GitHub


Images from CY 2021 are still available from this repository. These images are built with different backends:

  • Eigen;
  • oneDNN, with Compute Library for the Arm Architecture (ACL).

TensorFlow-Eigen (r21.12-tf-2.7.0-eigen) is the recommended image while TensorFlow-oneDNN-acl (r21.12-tf-2.7.0-onednn-acl) is experimental and any feedback is welcomed.

TagTensorFlow versiononeDNN versionCompute Library versionDescription
r21.12-tf-2.7.0-eigen2.7.0N/AN/A(Recommended) TensorFlow 2.7 is compiled with the default Eigen backend.
r21.12-tf-2.7.0-onednn-acl2.7.02.421.11(Experimental) TensorFlow 2.7 is compiled with oneDNN optimized with ACL.

Please refer to https://github.com/ARM-software/Tool-Solutions/tree/master/docker/tensorflow-aarch64 for the build recipes, examples on how to use the image and to report any issues.

For more information on Compute Library, refer to https://github.com/ARM-software/ComputeLibrary.

Docker Pull Command

docker pull armswdev/tensorflow-arm-neoverse-n1