diff --git a/finetune/Dockerfile.ldh b/finetune/Dockerfile.ldh index 8c6ba64..ecfe16d 100644 --- a/finetune/Dockerfile.ldh +++ b/finetune/Dockerfile.ldh @@ -78,17 +78,16 @@ RUN \ # Mellanox OFED -# ENV MLNX_OFED_VERSION=4.9-7.1.0.0 +ENV MLNX_OFED_VERSION=23.10-3.2.2.0 RUN \ source /opt/conda/etc/profile.d/conda.sh && conda activate ${CONDA_ENV_NAME} && \ apt-get install -y libnuma-dev libnvidia-compute-515 && \ cd ${STAGE_DIR} && \ - wget -q -O - https://content.mellanox.com/ofed/MLNX_OFED-5.8-4.1.5.0/MLNX_OFED_LINUX-5.8-4.1.5.0-ubuntu22.04-x86_64.tgz | tar xzf - && \ - cd MLNX_OFED_LINUX-5.8-4.1.5.0-ubuntu22.04-x86_64 && \ + wget -q -O - http://www.mellanox.com/downloads/ofed/MLNX_OFED-${MLNX_OFED_VERSION}/MLNX_OFED_LINUX-${MLNX_OFED_VERSION}-ubuntu22.04-x86_64.tgz | tar xzf - && \ + cd MLNX_OFED_LINUX-${MLNX_OFED_VERSION}-ubuntu22.04-x86_64 && \ ./mlnxofedinstall --user-space-only --without-fw-update --all -q && \ cd ${STAGE_DIR} && \ - rm -rf ${STAGE_DIR}/MLNX_OFED_LINUX-5.8-4.1.5.0-ubuntu22.04-x86_64.tgz* - + rm -rf ${STAGE_DIR}/MLNX_OFED_LINUX-${MLNX_OFED_VERSION}-ubuntu22.04-x86_64* # nv_peer_mem ENV NV_PEER_MEM_VERSION=1.2