From 7f2618b5f4eb4c8b2969182e80ce373c2017806a Mon Sep 17 00:00:00 2001 From: Wing Lian Date: Sat, 7 Oct 2023 12:23:31 -0400 Subject: [PATCH] add docker images for pytorch 2.10 (#697) --- .github/workflows/base.yml | 5 +++++ .github/workflows/main.yml | 10 ++++++++++ 2 files changed, 15 insertions(+) diff --git a/.github/workflows/base.yml b/.github/workflows/base.yml index d755b63263..5f08854842 100644 --- a/.github/workflows/base.yml +++ b/.github/workflows/base.yml @@ -25,6 +25,11 @@ jobs: python_version: "3.10" pytorch: 2.0.1 torch_cuda_arch_list: "7.0 7.5 8.0 8.6 9.0+PTX" + - cuda: "118" + cuda_version: 11.8.0 + python_version: "3.10" + pytorch: 2.1.0 + torch_cuda_arch_list: "7.0 7.5 8.0 8.6 9.0+PTX" steps: - name: Checkout uses: actions/checkout@v3 diff --git a/.github/workflows/main.yml b/.github/workflows/main.yml index a5b4d30379..5fdd2d705d 100644 --- a/.github/workflows/main.yml +++ b/.github/workflows/main.yml @@ -23,6 +23,11 @@ jobs: python_version: "3.10" pytorch: 2.0.1 axolotl_extras: + - cuda: 118 + cuda_version: 11.8.0 + python_version: "3.10" + pytorch: 2.1.0 + axolotl_extras: runs-on: [self-hosted, gpu, docker] steps: - name: Checkout @@ -68,6 +73,11 @@ jobs: pytorch: 2.0.1 axolotl_extras: is_latest: true + - cuda: 118 + cuda_version: 11.8.0 + python_version: "3.10" + pytorch: 2.1.0 + axolotl_extras: runs-on: [self-hosted, gpu, docker] steps: - name: Checkout