From 35f1fe8245209858720122fa6c79d387bb33ce26 Mon Sep 17 00:00:00 2001 From: Will Constable Date: Tue, 14 May 2024 20:31:49 -0700 Subject: [PATCH] try with faster docker ghstack-source-id: f0218c37ec0895dc585443784a415a0ec7757f9a Pull Request resolved: https://github.com/pytorch/torchtitan/pull/329 --- .github/workflows/unit_test_4gpu.yaml | 9 +++++---- 1 file changed, 5 insertions(+), 4 deletions(-) diff --git a/.github/workflows/unit_test_4gpu.yaml b/.github/workflows/unit_test_4gpu.yaml index ee426b91..84bc82ec 100644 --- a/.github/workflows/unit_test_4gpu.yaml +++ b/.github/workflows/unit_test_4gpu.yaml @@ -17,13 +17,14 @@ 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 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