Skip to content

Files

Latest commit

a714393 · Oct 24, 2023

History

History

Example4

Folders and files

NameName
Last commit message
Last commit date

parent directory

..
Oct 24, 2023
Oct 24, 2023
Oct 24, 2023
Oct 24, 2023

Purpose

Show how to use multiple GPUs with Tensorflow by modifying the example tf_test.py

Contents

  • tf_test_multi_gpu.py: Modified code tf_test.py to use all available GPUs on a node
  • run.sbatch: Slurm batch-job submission script to pull singularity image and run tf_test_multi_gpu.py
  • tf_test.out: Output file

Important notes

  1. In this example the slurm batch script pulls a singularity container with TensorFlow and runs the examples inside the singularity container. However, you can modify run.sbatch script to run within a conda/mamba environment.