#!/bin/bash
#SBATCH --job-name=TensorFlow-2.17.0-case
#SBATCH --partition=batch
#SBATCH --time=10:00

module load gcc/14.2.0 cuda/12.6.2 python/3.11 cudnn
source ~/tensorflow-cpu-env/bin/activate

# code to be executed
python3 tensorflow-test.py