hoho / hoho_gpu.batch
jskvrna's picture
Improves training efficiency and configuration
fb55e87
raw
history blame
782 Bytes
#!/bin/bash
#SBATCH --nodes=1 # 1 node
#SBATCH --ntasks-per-node=1 # 1 tasks per node
#SBATCH --cpus-per-task=16 # 6 CPUS per task = 12 CPUS per node
#SBATCH --mem-per-cpu=10G # 8GB per CPU = 96GB per node
#SBATCH --time=24:00:00 # time limits: 1 hour
#SBATCH --error=hoho_gpu.err # standard error file
#SBATCH --output=hoho_gpu.out # standard output file
#SBATCH --partition=amdgpu # partition name
#SBATCH --mail-user=skvrnjan@fel.cvut.cz # where send info about job
#SBATCH --mail-type=ALL # what to send, valid type values are NONE, BEGIN, END, FAIL, REQUEUE, ALL
#SBATCH --gres=gpu:1
cd /mnt/personal/skvrnjan/hoho/
module purge
module load Python/3.10.8-GCCcore-12.2.0
module load CUDA/12.6.0
source /mnt/personal/skvrnjan/venvs/hoho/bin/activate
python train_pnet_cluster.py