Show how to use multiple GPUs with Tensorflow by modifying the example tf_test.py
tf_test_multi_gpu.py
: Modified codetf_test.py
to use all available GPUs on a noderun.sbatch
: Slurm batch-job submission script to pull singularity image and runtf_test_multi_gpu.py
tf_test.out
: Output file
- 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.