From 91d9797e564229fc5e94892c98f38ad080db1a7b Mon Sep 17 00:00:00 2001 From: Vasilis Vryniotis Date: Fri, 21 May 2021 10:47:16 +0100 Subject: [PATCH] Change CI cuda versions to 10.2 (#3869) * Change cuda versions. * changing cu_version * patching regenerate.py * more changes. --- .circleci/config.yml | 32 ++++++++++++++++---------------- .circleci/config.yml.in | 10 +++++----- .circleci/regenerate.py | 6 +++--- 3 files changed, 24 insertions(+), 24 deletions(-) diff --git a/.circleci/config.yml b/.circleci/config.yml index fe0c913674d..85a2e273de5 100644 --- 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 index 3471b776a2f..992985d8550 100644 --- 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 index 5fea42c5b47..db6b44c089a 100755 --- 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)