Skip to navigation

Environment Variables

View as Markdown

Configure NIM for BMD behavior at container startup using the following environment variables.

Required Variables

VariableRequiredDefaultDescription
NGC_API_KEYYes (for MACE auto-download)—NGC authentication key. Required for the container to download the bundled MACE model from NGC.

Configuration Variables

VariableTypeDefaultDescription
ALCHEMI_NIM_MODEL_TYPEstring"mace"Model type: mace, aimnet2, or tensornet.
ALCHEMI_NIM_MODEL_PATHstringnullPath to the model file or directory inside the container.
ALCHEMI_NIM_PBCbooltruePeriodic boundary conditions. If true, requests must include cell. If false, requests must not include cell.
ALCHEMI_NIM_BATCH_SIZEint0Batch size in atoms (0 enables auto-estimation at startup).
ALCHEMI_NIM_BATCH_SIZE_ESTIMATE_DATAstring"omat24"Dataset used for auto-estimation: omat24 (bulk materials), omol25 (molecules), or omc25 (molecular crystals).
ALCHEMI_NIM_DFT3_ENABLEDboolfalseEnable DFT-D3(BJ) dispersion correction.
ALCHEMI_NIM_DFT3_PARAMstring (JSON)nullJSON string containing custom DFT-D3(BJ) parameters (for example, {"s8": 0.3908, "a1": 0.5660, "a2": 3.1280}). Refer to the ALCHEMI Toolkit-Ops documentation for details.

Container Runtime Variables

VariableRequiredDefaultDescription
NIM_HTTP_API_PORTNo8000HTTP port for the inference API.
NIM_DISABLE_MODEL_DOWNLOADNofalseWhether to automatically download the default MACE model from NGC.

For launch examples, refer to Custom Models and Quickstart Guide.