File size: 314 Bytes
ae02605
 
 
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
#!/bin/bash
#SBATCH -J benchH
#SBATCH -p gpu-h100
#SBATCH --gpus=h100:1
#SBATCH -c 4
#SBATCH --mem=16g
#SBATCH -t 0:20:00
#SBATCH -o logs/benchH-%j.out
source /etc/profile
SCR=/scratch/zt1/project/msml612/user/$USER
module load python/3.10.10
export PYTHONPATH=
cd $SCR/pcmd/src
$SCR/hf_env/bin/python -u bench.py