Modules
Lmod
module avail # list all available modules
module avail torch # search by name
module load torch/2.7 # load a module
module list # show loaded modules
module unload torch/2.7 # unload a module
module purge # unload everything#!/bin/bash
#SBATCH --job-name=torch-module
#SBATCH --nodes=2
#SBATCH --ntasks-per-node=1
#SBATCH --gpus-per-node=8
#SBATCH --cpus-per-task=48
#SBATCH --time=02:00:00
module load torch/2.7
srun python -m torch.distributed.run \
--nproc_per_node 8 \
--nnodes $SLURM_NNODES \
train.pySHPC (Singularity HPC)
Last updated

