From cc19588a37dc260c0a6e6fc6495589c957674538 Mon Sep 17 00:00:00 2001 From: Sandipan Roy Date: Wed, 12 Jun 2024 00:23:33 +0530 Subject: [PATCH] Need Fix for CVE-2024-5480 Need to fix CVE-2024-5480 as its a critical issue got pytorch. Before marge please test all functionality and changes required to merge it in main branch. Thanks --- training/deepspeed/Containerfile | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/training/deepspeed/Containerfile b/training/deepspeed/Containerfile index 854864d..2b86e18 100644 --- a/training/deepspeed/Containerfile +++ b/training/deepspeed/Containerfile @@ -3,7 +3,7 @@ FROM nvcr.io/nvidia/cuda:12.1.1-cudnn8-devel-ubi9 RUN dnf install -y python python-devel git RUN python -m ensurepip --upgrade -RUN pip3 install torch==2.1.2 --index-url https://download.pytorch.org/whl/cu121 +RUN pip3 install torch==2.3.1 --index-url https://download.pytorch.org/whl/cu121 RUN pip3 install packaging wheel RUN pip3 install flash-attn==2.5.7 RUN pip3 install deepspeed==0.14.2