#!/bin/sh
#SBATCH -p gpu
#SBATCH -n1
#SBATCH --gres gpu

export PATH=$HOME/miniconda3/bin:$PATH
export CONDA_PREFIX=$HOME/miniconda3/envs/gpu_env
export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:$CONDA_PREFIX/lib/
. activate base
conda activate gpu_env


python3 test_tf.py