Prerequisites

ENERGIZE requires Python and CUDA. The versions are specified in the requirements.txt and environment.yml files. It also requires the following Python modules:

• Pytorch

• Torchvision

• Torchmetrics

• Scikit-learn

• Dill

• Jsonschema

• PyYAML

• pyJoules[nvidia]