Skip to main content

This package contains Habana® TensorFlow bridge that allows to work with Habana® Gaudi®

Project description

Habana TensorFlow Python package

habana-tensorflow is a package designed for easy integration of Gaudi bridge to Tensorflow topologies. Main entry point (Python import) is habana_frameworks.tensorflow module that contains all the necessary functions to work with Gaudi.

Structure

Properly built wheel contains:

  • habana_frameworks python namespace (with all the folder structure inside)
  • libraries for all supported versions of TF (see habana_frameworks.tensorflow.sysconfig._supported_tf_versions) located in habana_frameworks/tensorflow/tf<version_with_underscores>/lib/habanalabs/
  • proper licensing

When loading habana_tensorflow, there is sanity check for currently installed TF version in the system, to match libraries found in the module.

TensorFlow package (habana_frameworks.tensorflow)

Most important module inside is library_loader and function load_habana_module(), that is exposed directly after import. It initializes the module properly.

Example:

import habana_frameworks.tensorflow as htf
htf.load_habana_module()

List of modules inside habana_frameworks.tensorflow

All modules are described in official docs.

Project details


Download files

Download the file for your platform. If you're not sure which to choose, learn more about installing packages.

Source Distributions

No source distribution files available for this release.See tutorial on generating distribution archives.

Built Distributions

habana_tensorflow-1.8.0.690-cp310-cp310-manylinux_2_17_x86_64.manylinux2014_x86_64.whl (13.2 MB view hashes)

Uploaded CPython 3.10 manylinux: glibc 2.17+ x86-64

habana_tensorflow-1.8.0.690-cp38-cp38-manylinux_2_17_x86_64.manylinux2014_x86_64.whl (13.2 MB view hashes)

Uploaded CPython 3.8 manylinux: glibc 2.17+ x86-64

Supported by

AWS AWS Cloud computing and Security Sponsor Datadog Datadog Monitoring Fastly Fastly CDN Google Google Download Analytics Microsoft Microsoft PSF Sponsor Pingdom Pingdom Monitoring Sentry Sentry Error logging StatusPage StatusPage Status page