From 3319780300b1cb9fa361b542f12b5c42f088c3f3 Mon Sep 17 00:00:00 2001
From: Wing Lian <wing.lian@gmail.com>
Date: Wed, 15 May 2024 09:45:27 -0400
Subject: [PATCH] update torch 2.2.1 -> 2.2.2 (#1622)

---
 .github/workflows/base.yml      | 2 +-
 .github/workflows/main.yml      | 4 ++--
 .github/workflows/nightlies.yml | 4 ++--
 .github/workflows/tests.yml     | 2 +-
 README.md                       | 2 +-
 5 files changed, 7 insertions(+), 7 deletions(-)

diff --git a/.github/workflows/base.yml b/.github/workflows/base.yml
index 2e8db61cc..4c1b0463a 100644
--- a/.github/workflows/base.yml
+++ b/.github/workflows/base.yml
@@ -30,7 +30,7 @@ jobs:
           - cuda: "121"
             cuda_version: 12.1.0
             python_version: "3.11"
-            pytorch: 2.2.1
+            pytorch: 2.2.2
             torch_cuda_arch_list: "7.0 7.5 8.0 8.6 8.7 8.9 9.0+PTX"
           - cuda: "121"
             cuda_version: 12.1.0
diff --git a/.github/workflows/main.yml b/.github/workflows/main.yml
index f51c42a82..99190dc71 100644
--- a/.github/workflows/main.yml
+++ b/.github/workflows/main.yml
@@ -28,7 +28,7 @@ jobs:
           - cuda: 121
             cuda_version: 12.1.0
             python_version: "3.11"
-            pytorch: 2.2.1
+            pytorch: 2.2.2
             axolotl_extras:
           - cuda: 121
             cuda_version: 12.1.0
@@ -89,7 +89,7 @@ jobs:
           - cuda: 121
             cuda_version: 12.1.0
             python_version: "3.11"
-            pytorch: 2.2.1
+            pytorch: 2.2.2
             axolotl_extras:
           - cuda: 121
             cuda_version: 12.1.0
diff --git a/.github/workflows/nightlies.yml b/.github/workflows/nightlies.yml
index f4172fa40..f668e5f65 100644
--- a/.github/workflows/nightlies.yml
+++ b/.github/workflows/nightlies.yml
@@ -27,7 +27,7 @@ jobs:
           - cuda: 121
             cuda_version: 12.1.0
             python_version: "3.11"
-            pytorch: 2.2.1
+            pytorch: 2.2.2
             axolotl_extras:
           - cuda: 121
             cuda_version: 12.1.0
@@ -89,7 +89,7 @@ jobs:
           - cuda: 121
             cuda_version: 12.1.0
             python_version: "3.11"
-            pytorch: 2.2.1
+            pytorch: 2.2.2
             axolotl_extras:
           - cuda: 121
             cuda_version: 12.1.0
diff --git a/.github/workflows/tests.yml b/.github/workflows/tests.yml
index a53640e0b..8f25eddc3 100644
--- a/.github/workflows/tests.yml
+++ b/.github/workflows/tests.yml
@@ -82,7 +82,7 @@ jobs:
           - cuda: 121
             cuda_version: 12.1.0
             python_version: "3.11"
-            pytorch: 2.2.1
+            pytorch: 2.2.2
             num_gpus: 1
     steps:
       - name: Checkout
diff --git a/README.md b/README.md
index 0b45bb78b..23ac443ee 100644
--- a/README.md
+++ b/README.md
@@ -302,7 +302,7 @@ Write a job description in YAML as below:
 # dstack.yaml
 type: task
 
-image: winglian/axolotl-cloud:main-20240429-py3.11-cu121-2.2.1
+image: winglian/axolotl-cloud:main-20240429-py3.11-cu121-2.2.2
 
 env:
   - HUGGING_FACE_HUB_TOKEN