From a10da1caff183cf986975a06f5c7ffc4f300fb22 Mon Sep 17 00:00:00 2001 From: Wing Lian Date: Sat, 1 Jul 2023 00:29:07 -0400 Subject: [PATCH] 11.7.0 nvidia/cuda docker images are deprecated, move to 11.7.1 --- .github/workflows/base.yml | 2 +- .github/workflows/main.yml | 4 ++-- 2 files changed, 3 insertions(+), 3 deletions(-) diff --git a/.github/workflows/base.yml b/.github/workflows/base.yml index 623083db2..f3ad69570 100644 --- a/.github/workflows/base.yml +++ b/.github/workflows/base.yml @@ -26,7 +26,7 @@ jobs: pytorch: 2.0.0 axolotl_extras: - cuda: "117" - cuda_version: 11.7.0 + cuda_version: 11.7.1 python_version: "3.9" pytorch: 1.13.1 axolotl_extras: diff --git a/.github/workflows/main.yml b/.github/workflows/main.yml index 033199154..07f25cac6 100644 --- a/.github/workflows/main.yml +++ b/.github/workflows/main.yml @@ -30,7 +30,7 @@ jobs: pytorch: 2.0.0 axolotl_extras: gptq - cuda: cu117 - cuda_version: 11.7.0 + cuda_version: 11.7.1 python_version: "3.9" pytorch: 1.13.1 axolotl_extras: @@ -85,7 +85,7 @@ jobs: pytorch: 2.0.0 axolotl_extras: gptq - cuda: cu117 - cuda_version: 11.7.0 + cuda_version: 11.7.1 python_version: "3.9" pytorch: 1.13.1 axolotl_extras: