covid_pipeline.sge 306 Bytes
Newer Older
Chris Jewell's avatar
Chris Jewell committed
1
2
3
#!/bin/bash

#$ -S /bin/bash
Chris Jewell's avatar
Chris Jewell committed
4
#$ -P chicas
Chris Jewell's avatar
Chris Jewell committed
5
6
7
8
9
10
11
12
13
#$ -q gpu
#$ -l ngpus=1
# -l ncpus=2
#$ -l h_vmem=32G
#$ -l h_rt=12:00:00
#$ -j y
#$ -cwd

. /etc/profile
Chris Jewell's avatar
Chris Jewell committed
14
. $HOME/.bash_profile
Chris Jewell's avatar
Chris Jewell committed
15

Chris Jewell's avatar
Chris Jewell committed
16
module add cuda/11.0
Chris Jewell's avatar
Chris Jewell committed
17
18
19
20
21

export XLA_FLAGS="--xla_gpu_cuda_data_dir=$CUDA_HOME"

echo Args: "$@"

22
poetry run python -m covid.pipeline "$@"