#!/bin/bash
#SBATCH --partition=gpu
#SBATCH --exclusive
#SBATCH --ntasks=20
#SBATCH --ntasks-per-node=20
#SBATCH --mem=0
#SBATCH --output=%j.out
#SBATCH --error=%j.err
#SBATCH --time=00:30:00
#SBATCH --job-name=af-test

module load CUDA/11.3.1

eval "$(/home/van/Programs/miniconda3/bin/conda shell.bash hook)"
conda activate af2.2.0

python -c "import tensorflow as tf; \
           print(tf.__version__); \
	   print('Num GPUs:', len(tf.config.list_physical_devices('GPU'))); \
           print(tf.config.list_physical_devices('GPU'));"
