From 2b2367efbf3e76837ae764f0c260d6d34bec5628 Mon Sep 17 00:00:00 2001 From: pengwa Date: Wed, 27 Jul 2022 06:11:04 +0800 Subject: [PATCH] Fix orttraining-linux-gpu-ci-pipeline (fairscale dependency) (#12320) authored by: @pengwa --- .../ci_build/github/linux/docker/scripts/install_python_deps.sh | 2 +- .../docker/scripts/training/ortmodule/stage2/requirements.txt | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/tools/ci_build/github/linux/docker/scripts/install_python_deps.sh b/tools/ci_build/github/linux/docker/scripts/install_python_deps.sh index 73063eb29b..81fefb3857 100755 --- a/tools/ci_build/github/linux/docker/scripts/install_python_deps.sh +++ b/tools/ci_build/github/linux/docker/scripts/install_python_deps.sh @@ -63,7 +63,7 @@ if [ $DEVICE_TYPE = "gpu" ]; then ${PYTHON_EXE} -m pip install -r ${0/%install_python_deps.sh/training\/ortmodule\/stage2\/requirements.txt} else ${PYTHON_EXE} -m pip install -r ${0/%install_python_deps.sh/training\/ortmodule\/stage1\/requirements_torch${TORCH_VERSION}_rocm${ROCM_VER}\/requirements.txt} - ${PYTHON_EXE} -m pip install fairscale + ${PYTHON_EXE} -m pip install fairscale==0.4.6 fi fi fi diff --git a/tools/ci_build/github/linux/docker/scripts/training/ortmodule/stage2/requirements.txt b/tools/ci_build/github/linux/docker/scripts/training/ortmodule/stage2/requirements.txt index 8c48bc704a..1ddf5ad25f 100644 --- a/tools/ci_build/github/linux/docker/scripts/training/ortmodule/stage2/requirements.txt +++ b/tools/ci_build/github/linux/docker/scripts/training/ortmodule/stage2/requirements.txt @@ -7,4 +7,4 @@ h5py wget pytorch-lightning==1.6.0 deepspeed==0.3.15 -fairscale +fairscale==0.4.6