From 02f4d69c9d324a04dad72d29079698c22629be63 Mon Sep 17 00:00:00 2001
From: Fangjun Kuang <csukuangfj@gmail.com>
Date: Sat, 10 Feb 2024 20:25:02 +0800
Subject: [PATCH] install torch 2.2.0

---
 scripts/github_actions/install_torch.sh | 13 +++++++++++++
 1 file changed, 13 insertions(+)

diff --git a/scripts/github_actions/install_torch.sh b/scripts/github_actions/install_torch.sh
index 1209ed6..d071a02 100755
--- a/scripts/github_actions/install_torch.sh
+++ b/scripts/github_actions/install_torch.sh
@@ -185,6 +185,19 @@ case ${torch} in
         ;;
     esac
     ;;
+  2.2.*)
+    case ${cuda} in
+      11.8)
+        package="torch==${torch}+cu118"
+        url=https://download.pytorch.org/whl/torch_stable.html
+        ;;
+      12.1)
+        package="torch==${torch}"
+        # Leave it empty to use PyPI.
+        url=
+        ;;
+    esac
+    ;;
   *)
     echo "Unsupported PyTorch version: ${torch}"
     exit 1