Skip to content

tensorflow

Back to ROCm containers

Description

TensorFlow is an end-to-end open source platform for machine learning.

Homepage

https://www.tensorflow.org

Available Versions on RCAC Clusters

Cluster Versions
BELL 2.5-rocm4.2-dev, 2.7-rocm5.0-dev, 2.16.2-rocm6.3.1-py3.12-dev, 2.17.0-rocm6.3.1-py3.12-dev
NEGISHI 2.5-rocm4.2-dev, 2.7-rocm5.0-dev

Module

You can load the container module by:

module load rocmcontainers
module load tensorflow