From 96a5cdf04fe2df996635c49552dcac56eb44e3f8 Mon Sep 17 00:00:00 2001 From: Will Constable Date: Tue, 14 May 2024 20:08:03 -0700 Subject: [PATCH] fix ghstack-source-id: c8cec41dd09ad540d5c81897a994dd40420d6e67 Pull Request resolved: https://github.com/pytorch/torchtitan/pull/328 --- .github/workflows/unit_test_4gpu.yaml | 5 ++++- 1 file changed, 4 insertions(+), 1 deletion(-) diff --git a/.github/workflows/unit_test_4gpu.yaml b/.github/workflows/unit_test_4gpu.yaml index 506c1bb7c..ee426b914 100644 --- a/.github/workflows/unit_test_4gpu.yaml +++ b/.github/workflows/unit_test_4gpu.yaml @@ -21,11 +21,14 @@ jobs: repository: "pytorch/torchtitan" upload-artifact: "outputs" script: | + conda create -n "test" python=3.10 + conda activate test + conda install git pip config --user set global.progress_bar off + pip install --upgrade pip python -m pip install --force-reinstall --pre torch --index-url https://download.pytorch.org/whl/nightly/cu121 python -m pip install -r requirements.txt python -m pip install -r dev-requirements.txt - python -m pip install git python -m pip install git+https://github.com/pytorch/pippy python ./test_runner.py mv outputs artifacts-to-be-uploaded