From 7495b33f3462e9ee1ac1553f2850a8a1c75d9299 Mon Sep 17 00:00:00 2001
From: Yiyao Yang <87518590+YiyaoYang1@users.noreply.github.com>
Date: Wed, 18 Oct 2023 18:42:13 +0800
Subject: [PATCH] Add torch 2.1.0 checking in CI (#1389)

---
 .circleci/test.yml                     | 20 +++++++++++++-------
 .github/workflows/merge_stage_test.yml | 24 +++++++++++++++++++-----
 .github/workflows/pr_stage_test.yml    | 12 +++++-------
 3 files changed, 37 insertions(+), 19 deletions(-)

diff --git a/.circleci/test.yml b/.circleci/test.yml
index 78f8da28a9..f0718f11e3 100644
--- a/.circleci/test.yml
+++ b/.circleci/test.yml
@@ -137,16 +137,22 @@ jobs:
         type: string
       cuda:
         type: enum
-        enum: ["10.1", "10.2", "11.1", "11.7"]
+        enum: ["10.1", "10.2", "11.1", "11.7", "11.8"]
       cudnn:
         type: integer
         default: 7
     machine:
-      image: ubuntu-2004-cuda-11.4:202110-01
+      image: linux-cuda-11:default
       docker_layer_caching: true
-    resource_class: gpu.nvidia.small
+    resource_class: gpu.nvidia.small.multi
     steps:
       - checkout
+      - run:
+          name: Install nvidia-container-toolkit and Restart Docker
+          command: |
+            sudo apt-get update
+            sudo apt-get install -y nvidia-container-toolkit
+            sudo systemctl restart docker
       - run:
           name: Build Docker image
           command: |
@@ -261,8 +267,8 @@ workflows:
             - lint
       - build_cpu:
           name: maximum_version_cpu
-          torch: 2.0.0
-          torchvision: 0.15.1
+          torch: 2.1.0
+          torchvision: 0.16.0
           python: 3.9.0
           requires:
             - minimum_version_cpu
@@ -290,8 +296,8 @@ workflows:
             - hold
       - build_cuda:
           name: maximum_version_gpu
-          torch: 2.0.0
-          cuda: "11.7"
+          torch: 2.1.0
+          cuda: "11.8"
           cudnn: 8
           requires:
             - hold
diff --git a/.github/workflows/merge_stage_test.yml b/.github/workflows/merge_stage_test.yml
index 33d2c13f4a..81b973ef69 100644
--- a/.github/workflows/merge_stage_test.yml
+++ b/.github/workflows/merge_stage_test.yml
@@ -58,7 +58,7 @@ jobs:
     strategy:
       matrix:
         python-version: [3.7]
-        torch: [1.6.0, 1.7.1, 1.8.1, 1.9.1, 1.10.1, 1.11.0, 1.12.0, 1.13.0]
+        torch: [1.6.0, 1.7.1, 1.8.1, 1.9.1, 1.10.1, 1.11.0, 1.12.0, 1.13.0, 2.0.0, 2.1.0]
         include:
           - torch: 1.6.0
             torchvision: 0.7.0
@@ -79,6 +79,14 @@ jobs:
           - torch: 2.0.0
             torchvision: 0.15.1
             python-version: 3.8
+          - torch: 2.1.0
+            torchvision: 0.16.0
+            python-version: 3.8
+        exclude:
+          - torch: 2.0.0
+            python-version: 3.7
+          - torch: 2.1.0
+            python-version: 3.7
     steps:
       - uses: actions/checkout@v3
       - name: Set up Python ${{ matrix.python-version }}
@@ -228,8 +236,8 @@ jobs:
     runs-on: macos-latest
     strategy:
       matrix:
-        python-version: [3.7]
-        torch: [1.6.0, 1.8.1, 1.13.0]
+        python-version: [3.8]
+        torch: [1.6.0, 1.8.1, 1.13.0, 2.1.0]
         include:
           - torch: 1.6.0
             torchvision: 0.7.0
@@ -237,6 +245,8 @@ jobs:
             torchvision: 0.9.1
           - torch: 1.13.0
             torchvision: 0.14.0
+          - torch: 2.1.0
+            torchvision: 0.16.0
     steps:
       - uses: actions/checkout@v3
       - name: Set up Python 3.7
@@ -273,13 +283,17 @@ jobs:
       matrix:
         python-version: [3.7]
         platform: [cpu, cu111]
-        torch: [1.8.1]
-        torchvision: [0.9.1]
+        torch: [1.8.1, 2.1.0]
+        torchvision: [0.9.1, 0.16.0]
         include:
           - python-version: 3.8
             platform: cu117
             torch: 2.0.0
             torchvision: 0.15.1
+          - python-version: 3.8
+            platform: cu118
+            torch: 2.1.0
+            torchvision: 0.16.0
     steps:
       - uses: actions/checkout@v3
       - name: Set up Python ${{ matrix.python-version }}
diff --git a/.github/workflows/pr_stage_test.yml b/.github/workflows/pr_stage_test.yml
index 9ab31ebfe0..a3582932ab 100644
--- a/.github/workflows/pr_stage_test.yml
+++ b/.github/workflows/pr_stage_test.yml
@@ -136,15 +136,13 @@ jobs:
     runs-on: windows-2022
     strategy:
       matrix:
-        python-version: [3.7]
-        platform: [cpu, cu111]
-        torch: [1.8.1]
-        torchvision: [0.9.1]
+        torch: [2.1.0]
+        torchvision: [0.16.0]
         include:
           - python-version: 3.8
-            platform: cu117
-            torch: 2.0.0
-            torchvision: 0.15.1
+            platform: cu118
+            torch: 2.1.0
+            torchvision: 0.16.0
     steps:
       - uses: actions/checkout@v3
       - name: Set up Python ${{ matrix.python-version }}