Pytorch is an open source machine learning framework that accelerates the path from research prototyping to production deployment.
RocketCE includes container images for Pytorch v1.13.0 and are based on the packages available in channel rocketce of anaconda.org.
In this version of RocketCE, the container images are available for both CPU and GPU. Container images for Pytorch are available for python 3.8, 3.9 and 3.10 versions and these container images are supported for both Power 9 and Power 10 Systems of IBM with linux-ppc64le.
CPU based images are available in rocketce/pytorch-cpu. These images are pulled using the following commands
CPU based images are available in rocketce/pytorch. These images are pulled using the following commands
RocketCE container images can be run using the following commands
For more information on docker run arguments, refer to this document.
Container can be verified by the following command
Important packages available as part of these container are
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