...
Setonix can support Machine Learning workloads thanks to the large number of AMD GPUs installed on the system. AMD maintains a TensorFlow branch with added support for its GPUs. An official AMD container is also available but lacks both support for Cray MPI and some core Python packages, making it unusable on Setonix. For this reason, Pawsey developed its own TensorFlow container, which is installed on Setonix and available through the module system. The Pawsey TensorFlow container is the only supported way to run TensorFlow on Setonix.
Our TensorFlow container is publicly distributed on quay.io (external site). To pull it with docker, and build on top of it, you can use the command
$ docker pull quay.io/pawsey/tensorflow
| Column | ||
|---|---|---|
|
...