Tensorflow module has known issues

After June 2024 maintenance, the module tensorflow/rocm5.6-tf2.12 has shown some problems. If you are facing issues using this module, please raise a ticket to our helpdesk and, in the meantime, try the following temporary fix:

Before loading the tensorflow module, first swap pawseyenv module to the previous software stack as explained here: June 2024 Software Update - Important Information
After the swap, then load the tensorflow module

TensorFlow is an open-source software library for machine learning.By taking advantage of multiple GPU nodes, it is possible to run machine learning and deep learning tasks at scale. Pages in this section of the Pawsey documentation will cover details on how to do it.

Acknowledgments

The Pawsey staff would like to thank Joel Geoffrey Dunstan, a UWA student who worked with TensorFlow during an internship, for having shared his acquired expertise on how to perform distributed computations using the tool. His feedback was the right starting point that led to the creation of this tutorial.

Pages in this section: