srun --partition=short --ntasks=2 --gres=gpu:1 --time=02:00:00 --pty bash