Machine Learning

Overview

Many machine learning packages can utilize general purpose graphics processing units (GPGPUs).  If supported by the respective machine learning framework or application, code execution can be manyfold, often orders of magnitude, faster on GPU nodes compared to nodes without GPU devices.

Rivanna has several nodes that are equipped with GPU devices.  These nodes are available in the GPU partition.  Access to a GPU node and its GPU device(s) requires specific SLURM directives or command line options as described in the Jobs using a GPU Node section.

 

Applications

Several machine learning software packages are installed on Rivanna.  The most commonly used ones are:

Other less commonly used software packages include:

  • Dragonn
  • LightGBM
  • XGBoost