Skip to content

Latest commit

 

History

History
62 lines (51 loc) · 1.16 KB

tensorflow.rst

File metadata and controls

62 lines (51 loc) · 1.16 KB
Raw

Tensorflow

Introduction

TensorFlow is an end-to-end open source platform for machine learning. For more information, please check: Home page: https://www.tensorflow.org Docker: docker://rocm/tensorflow

Versions

  • 2.5-rocm4.2-dev

Commands

  • python
  • python3
  • python3.6
  • ipython
  • ipython3
  • bazel
  • estimator_ckpt_converter
  • horovodrun
  • import_pb_to_tensorboard
  • jupyter
  • saved_model_cli
  • tensorboard
  • tflite_convert
  • mpirun
  • mpiexec
  • ompi_info

Module

You can load the modules by:

module load rocmcontainers
module load tensorflow

Example job

Warning

Using #!/bin/sh -l as shebang in the slurm job script will cause the failure of some biocontainer modules. Please use #!/bin/bash instead.

To run tensorflow on our clusters:

#!/bin/bash
#SBATCH -A myallocation     # Allocation name
#SBATCH -t 1:00:00
#SBATCH -N 1
#SBATCH -n 1
#SBATCH --job-name=tensorflow
#SBATCH --mail-type=FAIL,BEGIN,END
#SBATCH --error=%x-%J-%u.err
#SBATCH --output=%x-%J-%u.out

module --force purge
ml rocmcontainers tensorflow