Using the NVIDIA Mellanox InfiniBand Drivers

The DGX software stack for Red Hat-derived operating systems does not include the Mellanox OpenFabrics Enterprise Distribution (MLNX_OFED) for Linux. This is to ensure that the MLNX_OFED driver is in sync with the Red Hat distribution kernel. This section describes how to download, install, and upgrade MLNX_OFED on systems that are running Red Hat Enterprise Linux.

Prerequisites

Installing the NVIDIA Mellanox InfiniBand Drivers

Perform the following steps to install MLNX_OFED on systems that do not yet have it installed. It is imperative that a validated MLNX_OFED version is used for the RHEL version that the DGX system is running.

Important

Running the dnf update command at any time to install the drivers can update the system to the latest Red Hat Enterprise Linux version.

  1. Determine which version of Red Hat Enterprise Linux is installed on the DGX system.

    cat /etc/redhat-release
    
  2. After referring to the NVIDIA DGX Software for Red Hat Enterprise Linux 8 Release Notes document, download the MLNX_OFED software bundle:

    1. Go to the Linux InfiniBand Drivers page, scroll down to the Download wizard, and then click the LTS Download tab.

      _images/mlnx_ofed-select-download.png
    2. At the MLNX_OFED Download Center matrix, choose

      • The version to install (you may need to select Archive Versions),

      • RHEL/CentOS (under OS Distribution), and

      • The relevant OS Distribution Version and Architecture.

      _images/mlnx_ofed-select.png
    3. Click the supported ISO/tgz package.

      To obtain the download link, accept the End User License Agreement.

  3. After downloading the correct MLNX_OFED software bundle, follow the latest installation instructions:

    1. Go to the MLNX_OFED Software Releases site and select the MLNX_OFED software version you downloaded.

    2. Click the User Guide link and then navigate to Installation > Installing MLNX_OFED.

      Follow the installation instructions.

      Note

      The system may report that additional software needs to be installed before performing the installation. If such a message appears, install the software and then retry installing the MLNX_OFED driver.

  4. If you intend to use NVIDIA GPUDirect Storage (GDS), enable the driver’s GDS support according to the instructions at https://docs.nvidia.com/gpudirect-storage/troubleshooting-guide/index.html#mofed-req-install.

  5. Install nvidia-mlnx-config.

    sudo dnf install -y nvidia-mlnx-config
    
  6. Install kernel headers and development packages for your kernel.

    These are needed for the ensuing DKMS compilation.

    sudo dnf install -y kernel-headers-$(uname -r) kernel-devel-$(uname -r)
    
  7. After installing the MLNX_OFED drivers, install the NVIDIA peer memory module.

    sudo dnf install -y nvidia-peer-memory-dkms
    

Note

nvidia-peer-memory-dkms version 1.2 or later, requires MOFED version 5.4-3.0.3.0 or later. Using MOFED versions that do not fit this criteria will result in a build failure of the nv_peer_mem DKMS module. For more information, see:

Note

While in-box drivers may be available, using the in-box drivers is not recommended as they provide lower performance than the official MLNX OFED drivers and they do not support the GPUDirectTM

RDMA feature. For more information on configuring the in-box drivers, see the following Red Hat Enterprise Linux documentation:

Updating the NVIDIA Mellanox InfiniBand Drivers

Perform the following steps to update MOFED on systems that already have it installed. The Mellanox InfiniBand Drivers in RPM packages are precompiled for a specific kernel version.

It is imperative to use the correct MOFED version for the RHEL version that the DGX system has been updated to. There is no need to uninstall the current MOFED first, because the mlnxofedinstall script automatically uninstalls any previously installed versions.

  1. Upgrade the Red Hat Enterprise Linux release and kernel version.

    sudo dnf update --nobest
    
  2. Determine which version of Red Hat Enterprise Linux is installed on the DGX system.

    cat /etc/redhat-release
    
  3. After referring to the NVIDIA DGX Software for Red Hat Enterprise Linux 8 Release Notes document, download the MLNX_OFED software bundle:

    1. Go to the Linux InfiniBand Drivers page, scroll down to the Download wizard, and then click the LTS Download tab.

      _images/mlnx_ofed-select-download.png
    2. At the MLNX_OFED Download Center matrix, choose

      • The version to install (you may need to select Archive Versions),

      • RHEL/CentOS (under OS Distribution), and

      • The relevant OS Distribution Version and Architecture.

      _images/mlnx_ofed-select.png
    3. Click the supported ISO/tgz package.

      To obtain the download link, accept the End User License Agreement.

  4. Mount the downloaded ISO somewhere on the system.

    The following example shows the ISO being mounted on the /mnt directory.

    sudo mount MLNX_OFED_LINUX-<version>.iso /mnt
    
  5. Prepare to install the driver.

    1. Remove nvidia-mlnx-config and nvidia-peer-memory-dkms.

      sudo dnf remove -y nvidia-mlnx-config nvidia-peer-memory-dkms
      

      The mlnxofedinstall step will remove packages prior to installing new ones, causing nvidia-mlnx-config and nvidia-peer-memory-dkms to fall out because they depend on some of these removed packages. Removing those components ahead of time avoids issues. These will be reinstalled as a final step.

    2. Specify the new kernel version to use when installing the driver.

      NEXTKERNEL=$(sudo grubby --default-kernel | sed 's/.*vmlinuz\-//g')
      
  6. Install the driver with the - k and - s flags to specify the new kernel version and kernel source path.

    sudo /mnt/mlnxofedinstall -k ${NEXTKERNEL} -s /lib/modules/${NEXTKERNEL}/build --force
    

    Note

    The system might report that additional software must be installed before performing the installation. If the message appears, install the software and then repeat this step.

  7. Reboot.

    sudo reboot
    
  8. Reinstall nvidia-mlnx-config and nvidia-peer-memory-dkms.

    sudo dnf install -y nvidia-mlnx-config nvidia-peer-memory-dkms
    
  9. If you intend to use NVIDIA GPUDirect Storage (GDS), enable the driver’s GDS support according to the instructions at https://docs.nvidia.com/gpudirect-storage/troubleshooting-guide/index.html#mofed-req-install.