We have been providing PyTorch on Baobab via module for a long time. This works fine, but the version we provide may be out of date from time to time. Are you stuck in this case? Not at all.
The easiest way is to run PyTorch from apptainer, a Docker like container made for hpc.
Nvidia provides several Docker images that you can run on the cluster if the version we provide doesn’t suit your needs.
Steps to do:
- convert the Docker image to Apptainer format (sif)
- create a
sbatchscript to run your newly created image with Apptainer
- launch your
sbatchon the cluster
The good news is that we have already created the sif image for PyTorch2 : you can use it from
We created a sbatch example to launch PyTorch2 using Apptainer and GPU on Baobab.