From b26b3afe522085768fb3ea1f06cde69ecad55274 Mon Sep 17 00:00:00 2001 From: Will Constable Date: Tue, 14 May 2024 20:43:28 -0700 Subject: [PATCH] try with faster docker ghstack-source-id: 54ec6d26596e5f2e8d1134ec8321aec105f9acaf Pull Request resolved: https://github.com/pytorch/torchtitan/pull/329 --- .github/workflows/unit_test_4gpu.yaml | 11 +++++++---- 1 file changed, 7 insertions(+), 4 deletions(-) diff --git a/.github/workflows/unit_test_4gpu.yaml b/.github/workflows/unit_test_4gpu.yaml index ee426b91..21e55eac 100644 --- a/.github/workflows/unit_test_4gpu.yaml +++ b/.github/workflows/unit_test_4gpu.yaml @@ -17,13 +17,16 @@ jobs: gpu-arch-type: cuda gpu-arch-version: "12.1" # This image is faster to clone than the default, but it lacks CC needed by triton - # docker-image: "pytorch/pytorch:2.3.0-cuda12.1-cudnn8-runtime" + # (1m25s vs 2m37s) + docker-image: "pytorch/pytorch:2.3.0-cuda12.1-cudnn8-runtime" repository: "pytorch/torchtitan" upload-artifact: "outputs" + # conda create -n "test" python=3.10 + # conda activate test script: | - conda create -n "test" python=3.10 - conda activate test - conda install git + conda install -y -q git clang clangxx cxx-compiler + export CC=clang + export CXX=clangxx 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