PyTorch#

PyTorch can be run in batch, interactive, or Jupyter Notebook. For more information, check the module help information with module help pytorch.

PyTorch job#

The following example will use PyTorch to train a network on the MNIST data set.

First, download the PyTorch examples:

git clone https://github.com/pytorch/examples.git

Now that you have the examples, use the following job script to train the network.

pytorch-mnist.slurm

#!/bin/bash
#SBATCH --job-name=pytorch
#SBATCH --ntasks=1
#SBATCH --time=01:00:00
#SBATCH --account={PI_NetID}
#SBATCH --partition=gpu
#SBATCH --gres=gpu:1
#SBATCH --output=%x-%j.out

module load pytorch
python examples/mnist/main.py >> output.log

Submit the job:

cd /scratch/g/pi_netid/pytorch-test && sbatch pytorch.sh

PyTorch Jupyter Notebook#

This functionality is now provided by Open OnDemand! Use the Jupyter Notebook app to start a PyTorch enabled kernel.