gradio.sbatch

#!/bin/bash
#SBATCH --job-name="gradio_app_test"
#SBATCH --cpus-per-task=4
#SBATCH --gres=gpu:1,VRAM:24G
#SBATCH --constraint="GPU_MODEL:rtx_6000"
#SBATCH --mem=30G
#SBATCH --time=96:00:00
#SBATCH --mail-type=FAIL,INVALID_DEPEND,STAGE_OUT,TIME_LIMIT,REQUEUE
#SBATCH --output=/home/stud/yaji/slurm/logs/slurm-%j.out
#SBATCH --error=/home/stud/yaji/slurm/logs/slurm-%j.out

GPU_TYPE=$(nvidia-smi -i 0 --query-gpu=name --format=csv,noheader)
GPU_TYPE="${GPU_TYPE// /_}"
echo $GPU_TYPE

python app.py