#!/bin/sh

#SBATCH --job-name="GPU PyTorch"
#SBATCH --partition=main
#SBATCH --gpus=1
#SBATCH --nodes=1
#SBATCH --ntasks-per-node=1
#SBATCH --mem=20G
#SBATCH --time=00-00:25:00
#SBATCH -o slurm.%N.%j.out
#SBATCH -e slurm.%N.%j.err

apptainer exec docker://pytorch/pytorch:latest python example-03-gpu.py