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 inhabana_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
Release history Release notifications | RSS feed
Download files
Download the file for your platform. If you're not sure which to choose, learn more about installing packages.
Source Distributions
Built Distributions
Hashes for habana_tensorflow-1.7.1.85-cp310-cp310-manylinux_2_17_x86_64.manylinux2014_x86_64.whl
Algorithm | Hash digest | |
---|---|---|
SHA256 | 677cf610b12c546f77c3730c76e42142d371252750b2ffa8f02538475184adeb |
|
MD5 | 1807d13bc6e3c598b817603af75e2a71 |
|
BLAKE2b-256 | 67d4c96d0872705c84baaf5e9b5900e5d1298dd5922314189e8b4e1bfc7eed13 |
Hashes for habana_tensorflow-1.7.1.85-cp38-cp38-manylinux_2_17_x86_64.manylinux2014_x86_64.whl
Algorithm | Hash digest | |
---|---|---|
SHA256 | fa435405af8a6b35f92a1f84e1eb6e2ee3cc55c4103707637c4e04f3d4e3fc43 |
|
MD5 | fee2a7c1b711980cd5fae4bc58d094b4 |
|
BLAKE2b-256 | afb4c57e224b6e6d422f4bc9051166e67828fa133eb387b1380e40cc79ba3e50 |