diff --git a/defaults/main.yml b/defaults/main.yml
index 1f7f46f..eefa5c6 100644
--- a/defaults/main.yml
+++ b/defaults/main.yml
@@ -38,6 +38,11 @@ infiniband_upgrade: False
 # Name of the kernel headers package
 infiniband_kernel_headers_package: 'linux-headers'
 
+# Flavor of mlnx-ofed-* to install.
+# Values can be `basic`, `all`, `hpc`, etc.
+# See NVIDIA MLNX_OFED documentation for a complete list of flavors.
+infiniband_mlnx_ofed_flavor: 'basic'
+
 # Default to configure the repos and install the kernel modules.
 #
 # Some appliances like the NVIDIA DGX run their own software stack. When an
diff --git a/tasks/install-Debian.yml b/tasks/install-Debian.yml
index c399dc7..341b96c 100644
--- a/tasks/install-Debian.yml
+++ b/tasks/install-Debian.yml
@@ -2,8 +2,7 @@
 - name: Install MLNX_OFED packages for Debian OS family
   ansible.builtin.package:
     name:
-      - mlnx-ofed-kernel-dkms
-      - mlnx-ofed-kernel-utils
+      - "mlnx-ofed-{{ infiniband_mlnx_ofed_flavor }}"
       - "{{ infiniband_kernel_headers_package }}-{{ ansible_facts['kernel'] }}"
     state: "{{ 'latest' if infiniband_upgrade else 'present' }}"
   register: packages_install
diff --git a/tasks/install-RedHat.yml b/tasks/install-RedHat.yml
index 0a4c893..449731f 100644
--- a/tasks/install-RedHat.yml
+++ b/tasks/install-RedHat.yml
@@ -2,7 +2,6 @@
 - name: Install MLNX_OFED packages for Red Hat OS family
   ansible.builtin.package:
     name:
-      - mlnx-ofa_kernel-devel
-      - kmod-mlnx-ofa_kernel
+      - "mlnx-ofed-{{ infiniband_mlnx_ofed_flavor }}"
     state: "{{ 'latest' if infiniband_upgrade else 'present' }}"
   notify: "Ensure openibd.service is started"