#!/bin/bash

#SBATCH --time=1:00:00   # walltime.  hours:minutes:seconds
#SBATCH --ntasks=8   # number of processor cores (i.e. tasks)
#SBATCH --nodes=1   # number of nodes
#SBATCH --gpus=1
#SBATCH --mem=80G   # 164G memory per CPU core
#SBATCH --mail-user=aw742@byu.edu   # email address
#SBATCH --mail-type=BEGIN
#SBATCH --mail-type=END
#SBATCH --mail-type=FAIL
#SBATCH --qos=cs
#SBATCH --partition=cs

# some helpful debugging options
set -e
set -u

# LOAD MODULES, INSERT CODE, AND RUN YOUR PROGRAMS HERE
# module load python/3.11

# pip install virtualenv

# python -m venv mse_env

source ./mse_env/Scripts/activate

pip uninstall transformers
pip uninstall torch torchvision torchaudio

cd math-evaluation-harness
pip install -r requirements.txt
cd ..

pip install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu128
# pip install tqdm
# pip install accelerate