| #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 [email protected] # 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 |