RocketCE for Power

 View Only

Releasing RocketCE Tensorflow v2.10.1 with OpenCE v1.8.0

  • 1.  Releasing RocketCE Tensorflow v2.10.1 with OpenCE v1.8.0

    ROCKETEER
    Posted 01-12-2023 00:01

    Overview 

    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.8.0. In this version of RocketCE, there is TensorFlow 2.10.1. 

    These packages are supported for both Power 9 and Power 10 Systems of IBM with linux-ppc64le. 


    CPU Support
     

    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.

     

    GPU Support 

    This version of Tensorflow will have support for GPU based operations. This can be used in Power 9 systems.

     

    Installation instructions 

    CPU Only Variant 

    To install Tensorflow built for CPU support run the following command:
    • conda install -c rocketce tensorflow-cpu=2.10.1
     

     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=2.10.1

    GPU Variant 

    To install Tensorflow built for GPU support run the following command:

    • conda install -c rocketce tensorflow=2.10.1=cuda11.2*    - For CUDA 11.2 version
    • conda install -c rocketce tensorflow=2.10.1=cuda11.4*    - For CUDA 11.4 version

     

     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=2.10.1=cuda11.2*         - For CUDA 11.2 version
    • Conda list tensorflow=2.10.1=cuda11.4*         - For CUDA 11.4 version


    TensorFlow conda packages
     

    Following packages are available in rocketce channel and gets installed with tensorflow-cpu / tensorflow package 

    1. tensorflow-base 
    2. _tensorflow_select 
    3. tensorboard 
    4. tensorflow-estimator 

    Following packages are available in rocketce channel and which you can install to support tensorflow operations
     
           1. tensorflow-text
           2. tensorflow-hub

    More information 
    This conda package of CPU version is using the Power 10 MMA capabilities. More information is available in Matrix-multiply assist (MMA) capabilities.

    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: 



    ------------------------------
    Rajesh Nukala
    Rocket Internal - All Brands
    ------------------------------