Return to Image List

ibmz-accelerated-for-pytorch

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.


License

View license information here

As with all Docker images, these likely also contain other software which may be under other licenses (such as Bash, etc from the base distribution, along with any direct or indirect dependencies of the primary software being contained).

As for any pre-built image usage, it is the image user's responsibility to ensure that any use of this image complies with any relevant licenses for all software contained within.


Versions

Use the pull string below for the version of this image you require.
1.1.0 docker pull icr.io/ibmz/ibmz-accelerated-for-pytorch@sha256:92d4c27175829f22de6e8b7c0687766a6bc4b89617ac9148631c311addd0e8c0 Vulnerability Report11-14-2024
Version Pull String Security (IBM Cloud) Created

Usage Notes

For documentation and samples for the IBM Z Accelerated for Pytorch container image, please visit the GitHub Repository here.