TensorFlow is an open source library to help you develop and train machine learning models.
RocketCE includes several packages from the TensorFlow ecosystem and these packages are available in channel rocketce of anaconda.org.
First release of RocketCE is based on OpenCE Release v1.3.1. In this version of RocketCE, there is TensorFlow 2.5.1.
These packages are supported for both Power 9 and Power 10 Systems of IBM with linux-ppc64le.
RocketCE includes a version of TensorFlow built without GPU support. This inclusion allows for training and inferencing to be done on Power 9 and Power 10 systems that do not have GPUs, or on systems where you want to train and inference without using the GPUs.
This version of Tensorflow will not support GPU variant. Support of GPU variant will be provided in coming releases. For more information, refer main page of RocketCE.
CPU Only Variant
To install Tensorflow built for CPU support run the following command:
- conda install -c rocketce tensorflow-cpu
Once the installation is success, validate the installation by running the following command which lists the package tensorflow which is pulled from channel rocketce
- Conda list tensorflow-cpu
TensorFlow conda packages
Following packages are available in rocketce channel and gets installed with tensorflow-cpu package
The Tensorflow has various information, including tutorials, how to documents, and a getting started guide.
Additional tutorials and examples are available from the community, for example:
Rocket Internal - All Brands