[Fix] fix cudnn version of circle ci docker (#2069)

open-mmlab · Mar 15, 2023 · 3594f5d · 3594f5d
1 parent 4f5249d
commit 3594f5d
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 6 deletions.
diff --git a/.circleci/docker/Dockerfile b/.circleci/docker/Dockerfile
@@ -1,6 +1,6 @@
-ARG PYTORCH="1.8.1"
+ARG PYTORCH="1.7.1"
 ARG CUDA="11.1"
-ARG CUDNN="7"
+ARG CUDNN="8"
 
 FROM pytorch/pytorch:${PYTORCH}-cuda${CUDA}-cudnn${CUDNN}-devel
 

diff --git a/.circleci/test.yml b/.circleci/test.yml
@@ -83,10 +83,10 @@ jobs:
         type: string
       cuda:
         type: enum
-        enum: ["10.1", "10.2", "11.1"]
+        enum: ["11.1"]
       cudnn:
         type: integer
-        default: 7
+        default: 8
     machine:
       image: ubuntu-2004-cuda-11.4:202110-01
       # docker_layer_caching: true
@@ -168,10 +168,10 @@ workflows:
             - maximum_version_cpu
       - build_cuda:
           name: mainstream_version_gpu
-          torch: 1.8.1
+          torch: 1.7.1
           # Use double quotation mark to explicitly specify its type
           # as string instead of number
-          cuda: "10.2"
+          cuda: "11.1"
           requires:
             - hold
   merge_stage_test: