Support Matrix#

NVIDIA NIM for Batch Molecular Dynamics supports the following hardware and software.

CPU#

NIM for BMD requires an x86 processor. A modern processor is recommended for optimal performance.

Models#

Refer to Supported Models for the full list of supported model architectures, specific model variants, and sourcing details.

Supported Hardware#

The following table lists the supported and tested GPUs:

GPU

GPU Memory (GB)

Tested

NVIDIA RTX 6000 ADA

48

Yes

NVIDIA B200

192

Yes

NVIDIA H100 80GB HBM3

80

Yes

NVIDIA A100 80GB

80

Yes

NVIDIA L40S

48

Yes

Minimum requirements#

  • 8 GB GPU memory

  • CUDA compute capability >= 8.0

Any NVIDIA GPU meeting these requirements should work. Performance varies depending on the GPU model. Refer to Performance for benchmark results on specific GPUs.

Software#

NVIDIA Driver#

Refer to the Frameworks Support Matrix for NVIDIA driver support. Ensure that the latest compatible NVIDIA driver is installed on your system before launching NIM containers.

NVIDIA Container Toolkit#

Your Docker environment must support NVIDIA GPUs. For more information, refer to the NVIDIA Container Toolkit documentation.