TensorFlow
- Cristian Di Pietrantonio
- Alexis Espinosa
Default Tensorflow module with ROCm 5.6 is not officially supported
Pawsey strongly recommends to migrate Tensorflow slurm job scripts to make use of the new software stack available since March 2025. This because the currently installed GPU driver does not officially supports ROCm 5.6 and the default module (tensorflow/rocm5.6-tf2.12) use this now unsupported version.
The new software stack (March 2025) provides a new tensorflow module (tensorflow/2.13-rocm5.7.3) based on a supported version of ROCm.
- To make use of the new software stack please check instructions in the March 2025 Software Update - Important Information
- After swapping to the March 2025 software stack, you can proceed to load the new available tensorflow module
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.