Loading TensorFlow

  • moduletensorflow-cpu/latest
  • moduletensorflow/1.10
  • moduletensorflow/2.4.1
  • moduletensorflow/2.10
  • moduletensorflow-gpu/latest
  • moduletensorflow2-gpu/2.1.0
  • moduletensorflow2-gpu/2.2.0
  • moduletensorflow2-gpu/2.7.0
  • moduletensorflow-rocm/2.17.0
Tip: load modules inside your SLURM job script, not just on the login node, so compute nodes see the same environment. Verify with which after loading.

Example usage

Load a TensorFlow module

module load tensorflow/2.10

Availability

TensorFlow is available on: SeaWulf, NVwulf, ClinWulf.