File size: 895 Bytes
766ea9e |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 |
#!/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 |