blob: 718b6d2892cdeca7d77c06eb4334bfd5dacb28ca (
plain) (
blame)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
|
#!/bin/bash
#SBATCH --job-name=swr-teamprojekt
#SBATCH --partition=a100
#SBATCH --time=24:00:00
### Note: --gres=gpu:x should equal to ntasks-per-node
#SBATCH --nodes=1
#SBATCH --ntasks-per-node=1
#SBATCH --gres=gpu:a100:4
#SBATCH --cpus-per-task=16
#SBATCH --mem=32gb
#SBATCH --chdir=/mnt/lustre/mladm/mfa252/ref/
#SBATCH --output=/mnt/lustre/mladm/mfa252/%x-%j.out
source venv/bin/activate
### the command to run
srun ./hpc_train.sh
|
