Merge branch 'master' into master

pytorch · Apr 15, 2022 · 29fe185 · 29fe185
2 parents 03b4cfc + ca3f0b2
commit 29fe185
Showing 1 changed file with 8 additions and 2 deletions.
diff --git a/.circleci/config.yml b/.circleci/config.yml
@@ -61,7 +61,7 @@ setup_linux_system_environment: &setup_linux_system_environment
 
 pytorch_tutorial_build_defaults: &pytorch_tutorial_build_defaults
   machine:
-    image: ubuntu-1604-cuda-10.2:202012-01
+    image: ubuntu-2004-cuda-11.4:202110-01
   steps:
   - checkout
   - run:
@@ -75,7 +75,7 @@ pytorch_tutorial_build_defaults: &pytorch_tutorial_build_defaults
         sudo apt-get -y update
         sudo apt-get -y install expect-dev moreutils
 
-        sudo pip -q install awscli==1.16.35
+        sudo pip3 -q install awscli==1.16.35
 
          if [ -n "${CUDA_VERSION}" ]; then
            nvidia-smi
@@ -211,6 +211,7 @@ jobs:
 
   pytorch_tutorial_pr_build_worker_9:
     <<: *pytorch_tutorial_build_worker_defaults
+    resource_class: gpu.nvidia.small.multi
 
   pytorch_tutorial_pr_build_worker_10:
     <<: *pytorch_tutorial_build_worker_defaults
@@ -220,13 +221,15 @@ jobs:
 
   pytorch_tutorial_pr_build_worker_12:
     <<: *pytorch_tutorial_build_worker_defaults
+    resource_class: gpu.nvidia.small.multi
 
   pytorch_tutorial_pr_build_worker_13:
     <<: *pytorch_tutorial_build_worker_defaults
     resource_class: gpu.nvidia.small.multi
 
   pytorch_tutorial_pr_build_worker_14:
     <<: *pytorch_tutorial_build_worker_defaults
+    resource_class: gpu.nvidia.medium
 
   pytorch_tutorial_pr_build_worker_15:
     <<: *pytorch_tutorial_build_worker_defaults
@@ -277,6 +280,7 @@ jobs:
 
   pytorch_tutorial_master_build_worker_9:
     <<: *pytorch_tutorial_build_worker_defaults
+    resource_class: gpu.nvidia.small.multi
 
   pytorch_tutorial_master_build_worker_10:
     <<: *pytorch_tutorial_build_worker_defaults
@@ -286,13 +290,15 @@ jobs:
 
   pytorch_tutorial_master_build_worker_12:
     <<: *pytorch_tutorial_build_worker_defaults
+    resource_class: gpu.nvidia.small.multi
 
   pytorch_tutorial_master_build_worker_13:
     <<: *pytorch_tutorial_build_worker_defaults
     resource_class: gpu.nvidia.small.multi
 
   pytorch_tutorial_master_build_worker_14:
     <<: *pytorch_tutorial_build_worker_defaults
+    resource_class: gpu.nvidia.medium
 
   pytorch_tutorial_master_build_worker_15:
     <<: *pytorch_tutorial_build_worker_defaults