As of January 6, 2025, the LAS Pronto cluster was merged into the campus-wide Nova cluster. Information here may or may not work on Nova. Please contact Research IT if you have questions.

Machine Learning / GPU Container¶

As of July 2024, we discourage use of this container. Instead, please consider using a python virtual environment.

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/20230427.

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

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