From 7d23fc8fa2780fd5992ba40d6fe2d5a73403046b Mon Sep 17 00:00:00 2001 From: NanoCode012 Date: Thu, 22 Feb 2024 11:39:47 +0900 Subject: [PATCH] deprecate: pytorch 2.0.1 image (#1315) [skip ci] * deprecate: pytorch 2.0.1 image * deprecate from main image * Update main.yml * Update tests.yml --- .github/workflows/base.yml | 5 ----- .github/workflows/main.yml | 10 ---------- .github/workflows/tests.yml | 2 +- 3 files changed, 1 insertion(+), 16 deletions(-) diff --git a/.github/workflows/base.yml b/.github/workflows/base.yml index 07cf674737..381cf21ac1 100644 --- a/.github/workflows/base.yml +++ b/.github/workflows/base.yml @@ -12,11 +12,6 @@ jobs: fail-fast: false matrix: include: - - cuda: "118" - cuda_version: 11.8.0 - 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" diff --git a/.github/workflows/main.yml b/.github/workflows/main.yml index e699bc9acc..cbefb966a4 100644 --- a/.github/workflows/main.yml +++ b/.github/workflows/main.yml @@ -13,11 +13,6 @@ jobs: fail-fast: false matrix: include: - - cuda: 118 - cuda_version: 11.8.0 - python_version: "3.10" - pytorch: 2.0.1 - axolotl_extras: - cuda: 118 cuda_version: 11.8.0 python_version: "3.10" @@ -73,11 +68,6 @@ jobs: strategy: matrix: include: - - cuda: 118 - cuda_version: 11.8.0 - python_version: "3.10" - pytorch: 2.0.1 - axolotl_extras: - cuda: 118 cuda_version: 11.8.0 python_version: "3.10" diff --git a/.github/workflows/tests.yml b/.github/workflows/tests.yml index 2d99695241..802dbf0917 100644 --- a/.github/workflows/tests.yml +++ b/.github/workflows/tests.yml @@ -69,7 +69,7 @@ jobs: - cuda: 118 cuda_version: 11.8.0 python_version: "3.10" - pytorch: 2.0.1 + pytorch: 2.1.2 - cuda: 121 cuda_version: 12.1.0 python_version: "3.10"