PyTorch

PyTorch is a machine learning framework based on the Torch library, used for applications such as computer vision and natural language processing, originally developed by Meta AI and now part of the Linux Foundation umbrella.

PyTorch is available on Bridges-2 and Neocortex.

Documentation

• PyTorch home page

Using PyTorch on Bridges-2

PyTorch code development environments are available via the AI/pytorch modules on Bridges-2.  To see what is available and for more information, see the PSC created environments for AI applications page.

Using PyTorch on Neocortex

Your PyTorch code must be migrated to the Cerebras architecture before it will run successfully on Neocortex. You must port your code to use the Cerebras libraries instead of the default  libraries.

Detailed instructions to do this are given in the Code Migration section of the Neocortex User Documentation.