diff --git a/.circleci/config.yml b/.circleci/config.yml index c0c47d62e7a..4c7bf22dc67 100644 --- a/.circleci/config.yml +++ b/.circleci/config.yml @@ -1644,16 +1644,16 @@ workflows: name: cmake_linux_cpu python_version: '3.8' - cmake_linux_gpu: - cu_version: cu102 + cu_version: cu113 name: cmake_linux_gpu python_version: '3.8' - wheel_docker_image: pytorch/manylinux-cuda102 + wheel_docker_image: pytorch/manylinux-cuda113 - cmake_windows_cpu: cu_version: cpu name: cmake_windows_cpu python_version: '3.8' - cmake_windows_gpu: - cu_version: cu102 + cu_version: cu113 name: cmake_windows_gpu python_version: '3.8' - cmake_macos_cpu: diff --git a/.circleci/regenerate.py b/.circleci/regenerate.py index ac903fd0f71..2413e4adbac 100755 --- a/.circleci/regenerate.py +++ b/.circleci/regenerate.py @@ -268,9 +268,9 @@ def cmake_workflows(indentation=6): for device in device_types: job = {"name": f"cmake_{os_type}_{device}", "python_version": python_version} - job["cu_version"] = "cu102" if device == "gpu" else "cpu" + job["cu_version"] = "cu113" if device == "gpu" else "cpu" if device == "gpu" and os_type == "linux": - job["wheel_docker_image"] = "pytorch/manylinux-cuda102" + job["wheel_docker_image"] = "pytorch/manylinux-cuda113" jobs.append({f"cmake_{os_type}_{device}": job}) return indent(indentation, jobs)