ONNX Runtime is a cross-platform inference and training machine-learning accelerator.
ONNX Runtime is an accelerator for machine learning models with multi-platform support and a flexible interface to integrate with hardware-specific libraries. ONNX Runtime can be used with models from PyTorch, Tensorflow/Keras, TFLite, scikit-learn, and other frameworks.
This release of RocketCE is based on Release OpenCE v1.5.0 and it includes ONNX Runtime v1.10.0.
These packages are supported for both Power 9 and Power 10 Systems of IBM with linux-ppc64le.
RocketCE includes CPU only variant of ONNX Runtime which is 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 have support for GPU based operations. This can be used in Power 9 systems.
To install CPU only variant of ONNX Runtime, run one of the following commands according to the python versions you have:
Once the installation is success, validate the installation by running the following command which lists the package onnxruntime which was pulled from rocketce channel
To install GPU variant of ONNX Runtime, run one of the following commands according to the python versions you have:
77 4th AvenueWaltham, MA 02451 USA
Rocket Support Community
All Support Offerings
About Rocket Software
Training and Services
Forum Terms and Conditions
Contact Forum Moderator