TensorFlow
Tensorflow module has known issues
After June 2024 maintenance, the module tensorflow/rocm5.6-tf2.12
has shown some problems. For temporary fix check: June 2024 Software Update - Important Information
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.