From 2ed17b65eb357be5a8cc7485d91c86d0d3894162 Mon Sep 17 00:00:00 2001 From: epwalsh Date: Thu, 31 Oct 2024 12:52:42 -0700 Subject: [PATCH] add user-space drivers --- src/Dockerfile | 11 +++++++++++ 1 file changed, 11 insertions(+) diff --git a/src/Dockerfile b/src/Dockerfile index 9076efdf..545ab2f8 100644 --- a/src/Dockerfile +++ b/src/Dockerfile @@ -55,6 +55,17 @@ RUN apt-get update && apt-get install -y --no-install-recommends \ git && \ rm -rf /var/lib/apt/lists/* +# Install MLNX OFED user-space drivers +# See https://docs.nvidia.com/networking/pages/releaseview.action?pageId=15049785#Howto:DeployRDMAacceleratedDockercontaineroverInfiniBandfabric.-Dockerfile +ENV MOFED_VER 24.01-0.3.3.1 +ENV OS_VER ubuntu20.04 +ENV PLATFORM x86_64 +RUN wget --quiet https://content.mellanox.com/ofed/MLNX_OFED-${MOFED_VER}/MLNX_OFED_LINUX-${MOFED_VER}-${OS_VER}-${PLATFORM}.tgz && \ + tar -xvf MLNX_OFED_LINUX-${MOFED_VER}-${OS_VER}-${PLATFORM}.tgz && \ + MLNX_OFED_LINUX-${MOFED_VER}-${OS_VER}-${PLATFORM}/mlnxofedinstall --basic --user-space-only --without-fw-update -q && \ + rm -rf MLNX_OFED_LINUX-${MOFED_VER}-${OS_VER}-${PLATFORM} && \ + rm MLNX_OFED_LINUX-${MOFED_VER}-${OS_VER}-${PLATFORM}.tgz + # Install/upgrade Python build dependencies. RUN pip install --upgrade --no-cache-dir pip wheel packaging