From 19b0e900a405718df10cc2ce2be476a283df3b9a Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?=E8=B0=A2=E6=98=95=E8=BE=B0?= <xiexinch@outlook.com>
Date: Thu, 12 Oct 2023 04:33:00 -0500
Subject: [PATCH] [Dev] update CI config (#3377)

---
 .circleci/docker/Dockerfile |  2 ++
 .circleci/test.yml          | 34 +++++++++++++++++++++++++---------
 2 files changed, 27 insertions(+), 9 deletions(-)

diff --git a/.circleci/docker/Dockerfile b/.circleci/docker/Dockerfile
index b1d40e0e14..d014917af3 100644
--- a/.circleci/docker/Dockerfile
+++ b/.circleci/docker/Dockerfile
@@ -5,6 +5,8 @@ ARG CUDNN="7"
 
 FROM pytorch/pytorch:${PYTORCH}-cuda${CUDA}-cudnn${CUDNN}-devel
 
+ARG DEBIAN_FRONTEND=noninteractive
+
 # To fix GPG key error when running apt-get update
 RUN apt-key adv --fetch-keys https://developer.download.nvidia.com/compute/cuda/repos/ubuntu1804/x86_64/3bf863cc.pub
 RUN apt-key adv --fetch-keys https://developer.download.nvidia.com/compute/machine-learning/repos/ubuntu1804/x86_64/7fa2af80.pub
diff --git a/.circleci/test.yml b/.circleci/test.yml
index 57f89e18f5..b5d3e23a4d 100644
--- a/.circleci/test.yml
+++ b/.circleci/test.yml
@@ -82,16 +82,22 @@ jobs:
         type: string
       cuda:
         type: enum
-        enum: ["10.1", "10.2", "11.1"]
+        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
-      # docker_layer_caching: true
-    resource_class: gpu.nvidia.small
+      image: linux-cuda-11:default
+      docker_layer_caching: true
+    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:
           # Cloning repos in VM since Docker doesn't have access to the private key
           name: Clone Repos
@@ -154,9 +160,8 @@ workflows:
             - lint
       - build_cpu:
           name: maximum_version_cpu
-          # TODO: Fix torch 1.13 forward crush
-          torch: 1.12.0
-          torchvision: 0.13.0
+          torch: 2.0.0
+          torchvision: 0.15.0
           python: 3.9.0
           requires:
             - minimum_version_cpu
@@ -166,10 +171,10 @@ workflows:
             - maximum_version_cpu
       - build_cuda:
           name: mainstream_version_gpu
-          torch: 1.8.1
+          torch: 2.0.0
           # Use double quotation mark to explicitly specify its type
           # as string instead of number
-          cuda: "10.2"
+          cuda: "11.8"
           requires:
             - hold
   merge_stage_test:
@@ -188,3 +193,14 @@ workflows:
               only:
                 - dev-1.x
                 - main
+      - build_cuda:
+          name: maximum_version_gpu
+          torch: 2.0.0
+          # Use double quotation mark to explicitly specify its type
+          # as string instead of number
+          cuda: "11.8"
+          cudnn: 8
+          filters:
+            branches:
+              only:
+                - main