System Requirements#

  • Operating System

    • Ubuntu 24.04

    • Windows supported via WSL

  • Driver and GPU Requirements

    • pip: NVIDIA driver that is compatible with local PyTorch installation.

    • Docker container: PhysicsNeMo container release 25.06 is based on PyTorch 25.04 container. For driver requirements, please refer PyTorch NVIDIA Container Release Notes.

  • Required installations for pip install

    • Python >= 3.10

  • Recommended Hardware

    • 64-bit x86

    • NVIDIA GPUs:

      • Blackwell GPUs - B100

      • Hopper GPUs - H100

      • NVIDIA Ampere GPUs - A100, A30, A40, A4000, A6000

      • Turing GPUs - T4

    • Other Supported GPUs:

      • NVIDIA Ampere GPUs - RTX 30xx

    • For others, please reach us out at PhysicsNeMo Forums

    • 64-bit ARM architecture is also supported.

Note

To get the benefits of all the performance improvements (e.g. AMP, multi-GPU scaling, etc.), use the NVIDIA container for PhysicsNeMo. This container comes with all the prerequisites and dependencies and allows you to get started efficiently with PhysicsNeMo.