Division of Information Technology

TensorFlow is a Python library that uses GPU acceleration to complete linear algebra and machine learning tasks.  The TensorFlow libraries are available in Python through Anaconda. To use it to its fullest extent, place your TensorFlow job on the GPU queue by providing the -p flag in your slurm script:

#SBATCH -p gpu

There are several versions of TensorFlow available on SeaWulf. You can activate tensorflow with the following command:

module load tensorflow/2.4.1

You can activate a GPU-enabled TensorFlow with :

module load tensorflow2-gpu/2.2.0

These modules will activate a conda environment with several packages.

A hello world program for TensorFlow can be found in:

/gpfs/projects/samples/tensorflow

To make sure TensorFlow can access all 8 gpus on the node, you can run the following program:

/gpfs/projects/samples/tensorflow/tensor_list_gpus.py