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

. gpu_env/bin/activate
python3 test.py