Machine Learning / GPU Container

ResearchIT maintains a machine learning container (ml-gpu) that contains the CUDA drivers and has many machine learning frameworks available.

The singularity recipe that builds this container can be found here.

The latest version of the container is ml-gpu/20220603.

To start python within the container, run the following in an interactive session on a GPU node:

module load ml-gpu/20220603
ml-gpu python3 hello.py