How to use TensorFlow on Seawulf

Audience: Faculty, Researchers and Staff

This KB Article References: High Performance Computing
This Information is Intended for: Faculty, Researchers, Staff
Last Updated: December 13, 2023
Average Rating: Not Rated
Your feedback is important to us, help us by logging in to rate this article and provide feedback.

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

 

Submit a ticket

Additional Information


There are no additional resources available for this article.

Provide Feedback


Your feedback is important to us, help us by logging in to rate this article and provide feedback.

Sign in with NetID

Getting Help


The Division of Information Technology provides support on all of our services. If you require assistance please submit a support ticket through the IT Service Management system.

Submit A Quick Ticket

For More Information Contact


IACS Support System