#!/bin/bash
#SBATCH --job-name=rtx_example
#SBATCH --time=00:05:00
#SBATCH --nodes=1
#SBATCH --ntasks-per-node=1
#SBATCH --cpus-per-task=1
#SBATCH --mem-per-cpu=8GB
#SBATCH --partition=gpu
#SBATCH --gpus-per-node=rtx_pro_6000:1
#SBATCH --output=rtx_example.out

# Load virtual Python and venv
module load Python/3.13.1-GCCcore-14.2.0
source ~/venvs/rtx6000_venv/bin/activate

# Create a directory for the torchvision script
mkdir -p ~/torchvision_example
cd ~/torchvision_example

# Copy the script from the examples folder
cp /scratch/public/gpu_examples/main.py .

# Add execute permissions
chmod +x main.py

# Run the example
python3 main.py