PyTorch is an open source machine learning framework. It has a comprehensive set of tools that enable model development, training, and inference. It also features a rich, robust ecosystem.
On IBM® z16™ and later (running Linux on IBM Z or IBM® z/OS® Container Extensions (IBM zCX)), PyTorch will leverage new inference acceleration capabilities that target the IBM Integrated Accelerator for AI through the IBM z Deep Neural Network (zDNN) library. The IBM zDNN library contains a set of primitives that support Deep Neural Networks. These primitives transparently target the IBM Integrated Accelerator for AI on IBM z16 and later. No changes to the original model are needed to take advantage of the new inference acceleration capabilities.
Note: When using IBM Z Accelerated for PyTorch on either an IBM z15® or an IBM z14®, PyTorch will transparently target the CPU with no changes to the model.IBM Z Accelerated for Pytorch pytorch
See IBM Z Accelerated for Pytorch for more information This image is built by IBM to run on the IBM Z architecture and is not affiliated with any other community that provides a version of this image.1.1.0 | docker pull icr.io/ibmz/ibmz-accelerated-for-pytorch@sha256:92d4c27175829f22de6e8b7c0687766a6bc4b89617ac9148631c311addd0e8c0 | Vulnerability Report | 11-14-2024 |
Version | Pull String | Security (IBM Cloud) | Created |
---|---|---|---|
For documentation and samples for the IBM Z Accelerated for Pytorch container image, please visit the GitHub Repository here.