[fbsync] Change CI cuda versions to 10.2 (#3869)

Summary: * Change cuda versions. * changing cu_version * patching regenerate.py * more changes. Reviewed By: vincentqb, cpuhrsch Differential Revision: D28677174 fbshipit-source-id: a32861bd62e3f5a3d5b19106e4f1773128ba1006
pytorch · May 25, 2021 · b89cf06 · b89cf06
1 parent 908d35b
commit b89cf06
Show file tree

Hide file tree

Showing 3 changed files with 24 additions and 24 deletions.
diff --git a/.circleci/config.yml b/.circleci/config.yml
@@ -133,7 +133,7 @@ binary_common: &binary_common
     wheel_docker_image:
       description: "Wheel only: what docker image to use"
       type: string
-      default: "pytorch/manylinux-cuda101"
+      default: "pytorch/manylinux-cuda102"
     conda_docker_image:
       description: "Conda only: what docker image to use"
       type: string
@@ -660,10 +660,10 @@ jobs:
   unittest_linux_gpu:
     <<: *binary_common
     machine:
-      image: ubuntu-1604-cuda-10.1:201909-23
+      image: ubuntu-1604-cuda-10.2:202012-01
     resource_class: gpu.small
     environment:
-      image_name: "pytorch/manylinux-cuda101"
+      image_name: "pytorch/manylinux-cuda102"
       PYTHON_VERSION: << parameters.python_version >>
     steps:
       - checkout
@@ -743,7 +743,7 @@ jobs:
     executor:
       name: windows-gpu
     environment:
-      CUDA_VERSION: "10.1"
+      CUDA_VERSION: "10.2"
       PYTHON_VERSION: << parameters.python_version >>
     steps:
       - checkout
@@ -839,7 +839,7 @@ jobs:
   cmake_linux_gpu:
     <<: *binary_common
     machine:
-      image: ubuntu-1604-cuda-10.1:201909-23
+      image: ubuntu-1604-cuda-10.2:202012-01
     resource_class: gpu.small
     environment:
       PYTHON_VERSION: << parameters.python_version >>
@@ -1468,7 +1468,7 @@ workflows:
           name: unittest_linux_cpu_py3.9
           python_version: '3.9'
       - unittest_linux_gpu:
-          cu_version: cu101
+          cu_version: cu102
           filters:
             branches:
               only:
@@ -1477,7 +1477,7 @@ workflows:
           name: unittest_linux_gpu_py3.6
           python_version: '3.6'
       - unittest_linux_gpu:
-          cu_version: cu101
+          cu_version: cu102
           filters:
             branches:
               only:
@@ -1486,11 +1486,11 @@ workflows:
           name: unittest_linux_gpu_py3.7
           python_version: '3.7'
       - unittest_linux_gpu:
-          cu_version: cu101
+          cu_version: cu102
           name: unittest_linux_gpu_py3.8
           python_version: '3.8'
       - unittest_linux_gpu:
-          cu_version: cu101
+          cu_version: cu102
           filters:
             branches:
               only:
@@ -1515,7 +1515,7 @@ workflows:
           name: unittest_windows_cpu_py3.9
           python_version: '3.9'
       - unittest_windows_gpu:
-          cu_version: cu101
+          cu_version: cu102
           filters:
             branches:
               only:
@@ -1524,7 +1524,7 @@ workflows:
           name: unittest_windows_gpu_py3.6
           python_version: '3.6'
       - unittest_windows_gpu:
-          cu_version: cu101
+          cu_version: cu102
           filters:
             branches:
               only:
@@ -1533,11 +1533,11 @@ workflows:
           name: unittest_windows_gpu_py3.7
           python_version: '3.7'
       - unittest_windows_gpu:
-          cu_version: cu101
+          cu_version: cu102
           name: unittest_windows_gpu_py3.8
           python_version: '3.8'
       - unittest_windows_gpu:
-          cu_version: cu101
+          cu_version: cu102
           filters:
             branches:
               only:
@@ -1569,16 +1569,16 @@ workflows:
           name: cmake_linux_cpu
           python_version: '3.8'
       - cmake_linux_gpu:
-          cu_version: cu101
+          cu_version: cu102
           name: cmake_linux_gpu
           python_version: '3.8'
-          wheel_docker_image: pytorch/manylinux-cuda101
+          wheel_docker_image: pytorch/manylinux-cuda102
       - cmake_windows_cpu:
           cu_version: cpu
           name: cmake_windows_cpu
           python_version: '3.8'
       - cmake_windows_gpu:
-          cu_version: cu101
+          cu_version: cu102
           name: cmake_windows_gpu
           python_version: '3.8'
       - cmake_macos_cpu:

diff --git a/.circleci/config.yml.in b/.circleci/config.yml.in
@@ -133,7 +133,7 @@ binary_common: &binary_common
     wheel_docker_image:
       description: "Wheel only: what docker image to use"
       type: string
-      default: "pytorch/manylinux-cuda101"
+      default: "pytorch/manylinux-cuda102"
     conda_docker_image:
       description: "Conda only: what docker image to use"
       type: string
@@ -660,10 +660,10 @@ jobs:
   unittest_linux_gpu:
     <<: *binary_common
     machine:
-      image: ubuntu-1604-cuda-10.1:201909-23
+      image: ubuntu-1604-cuda-10.2:202012-01
     resource_class: gpu.small
     environment:
-      image_name: "pytorch/manylinux-cuda101"
+      image_name: "pytorch/manylinux-cuda102"
       PYTHON_VERSION: << parameters.python_version >>
     steps:
       - checkout
@@ -743,7 +743,7 @@ jobs:
     executor:
       name: windows-gpu
     environment:
-      CUDA_VERSION: "10.1"
+      CUDA_VERSION: "10.2"
       PYTHON_VERSION: << parameters.python_version >>
     steps:
       - checkout
@@ -839,7 +839,7 @@ jobs:
   cmake_linux_gpu:
     <<: *binary_common
     machine:
-      image: ubuntu-1604-cuda-10.1:201909-23
+      image: ubuntu-1604-cuda-10.2:202012-01
     resource_class: gpu.small
     environment:
       PYTHON_VERSION: << parameters.python_version >>

diff --git a/.circleci/regenerate.py b/.circleci/regenerate.py
@@ -234,7 +234,7 @@ def unittest_workflows(indentation=6):
                 if device_type == 'gpu':
                     if python_version != "3.8":
                         job['filters'] = gen_filter_branch_tree('master', 'nightly')
-                    job['cu_version'] = 'cu101'
+                    job['cu_version'] = 'cu102'
                 else:
                     job['cu_version'] = 'cpu'
 
@@ -255,9 +255,9 @@ def cmake_workflows(indentation=6):
                 'python_version': python_version
             }
 
-            job['cu_version'] = 'cu101' if device == 'gpu' else 'cpu'
+            job['cu_version'] = 'cu102' if device == 'gpu' else 'cpu'
             if device == 'gpu' and os_type == 'linux':
-                job['wheel_docker_image'] = 'pytorch/manylinux-cuda101'
+                job['wheel_docker_image'] = 'pytorch/manylinux-cuda102'
             jobs.append({f'cmake_{os_type}_{device}': job})
     return indent(indentation, jobs)