Versions Compared

Key

  • This line was added.
  • This line was removed.
  • Formatting was changed.


...

width900

...

titleTensorflow module has known issues

After June 2024 maintenance, the moduleĀ tensorflow/rocm5.6-tf2.12 has shown some problems. Most of them are fixed by defining the path for ROCM_PATH before loading the Tensorflow module:

$ export ROCM_PATH="/opt/rocm"
$ module load tensorflow/rocm5.6-tf2.12

...



.

Column


Note

This page is still a work in progress and support for Machine Learning workload has just started. Please check it frequently for updates.


...

FROM quay.io/pawsey/tensorflow:2.12.1.570-rocm5.6.0

To pull the image to your local desktop with Docker you can use:

$ docker pull quay.io/pawsey/tensorflow:2.12.1.570-rocm5.6.0

To know more about our recommendations of container builds with Docker and later translation into Singularity format for their use in Setonix please refer to the Containers Documentation.

...