From 334bd468a63787dc14974f41770efaca69299810 Mon Sep 17 00:00:00 2001 From: Nikita Shulga Date: Mon, 6 Feb 2023 20:01:41 +0000 Subject: [PATCH 1/2] [CI] Update CUDA to 11.7 As 11.6 builds has been disabled by https://github.com/pytorch/pytorch/pull/93406 --- .circleci/config.yml | 4 ++-- .circleci/config.yml.in | 4 ++-- 2 files changed, 4 insertions(+), 4 deletions(-) diff --git a/.circleci/config.yml b/.circleci/config.yml index 9519ca57541..b3f3eef23ca 100644 --- a/.circleci/config.yml +++ b/.circleci/config.yml @@ -142,14 +142,14 @@ pytorch_tutorial_build_defaults: &pytorch_tutorial_build_defaults pytorch_tutorial_build_worker_defaults: &pytorch_tutorial_build_worker_defaults environment: - DOCKER_IMAGE: "308535385114.dkr.ecr.us-east-1.amazonaws.com/pytorch/pytorch-linux-bionic-cuda11.6-cudnn8-py3-gcc7" + DOCKER_IMAGE: "308535385114.dkr.ecr.us-east-1.amazonaws.com/pytorch/pytorch-linux-bionic-cuda11.7-cudnn8-py3-gcc7" CUDA_VERSION: "9" resource_class: gpu.nvidia.small <<: *pytorch_tutorial_build_defaults pytorch_tutorial_build_manager_defaults: &pytorch_tutorial_build_manager_defaults environment: - DOCKER_IMAGE: "308535385114.dkr.ecr.us-east-1.amazonaws.com/pytorch/pytorch-linux-bionic-cuda11.6-cudnn8-py3-gcc7" + DOCKER_IMAGE: "308535385114.dkr.ecr.us-east-1.amazonaws.com/pytorch/pytorch-linux-bionic-cuda11.7-cudnn8-py3-gcc7" resource_class: medium <<: *pytorch_tutorial_build_defaults diff --git a/.circleci/config.yml.in b/.circleci/config.yml.in index a9bdffa1995..dd3b7242929 100644 --- a/.circleci/config.yml.in +++ b/.circleci/config.yml.in @@ -142,14 +142,14 @@ pytorch_tutorial_build_defaults: &pytorch_tutorial_build_defaults pytorch_tutorial_build_worker_defaults: &pytorch_tutorial_build_worker_defaults environment: - DOCKER_IMAGE: "308535385114.dkr.ecr.us-east-1.amazonaws.com/pytorch/pytorch-linux-bionic-cuda11.6-cudnn8-py3-gcc7" + DOCKER_IMAGE: "308535385114.dkr.ecr.us-east-1.amazonaws.com/pytorch/pytorch-linux-bionic-cuda11.7-cudnn8-py3-gcc7" CUDA_VERSION: "9" resource_class: gpu.nvidia.small <<: *pytorch_tutorial_build_defaults pytorch_tutorial_build_manager_defaults: &pytorch_tutorial_build_manager_defaults environment: - DOCKER_IMAGE: "308535385114.dkr.ecr.us-east-1.amazonaws.com/pytorch/pytorch-linux-bionic-cuda11.6-cudnn8-py3-gcc7" + DOCKER_IMAGE: "308535385114.dkr.ecr.us-east-1.amazonaws.com/pytorch/pytorch-linux-bionic-cuda11.7-cudnn8-py3-gcc7" resource_class: medium <<: *pytorch_tutorial_build_defaults {% raw %} From 1b9f6ddf01fcccff558bc1c81d359ca6c30797bd Mon Sep 17 00:00:00 2001 From: Nikita Shulga Date: Mon, 6 Feb 2023 20:29:30 +0000 Subject: [PATCH 2/2] And fix this as well --- .jenkins/get_docker_tag.py | 13 +++++++++---- 1 file changed, 9 insertions(+), 4 deletions(-) diff --git a/.jenkins/get_docker_tag.py b/.jenkins/get_docker_tag.py index 4d9db76b4a1..21c4a8f7089 100644 --- a/.jenkins/get_docker_tag.py +++ b/.jenkins/get_docker_tag.py @@ -5,9 +5,14 @@ } if __name__ == "__main__": - url = "https://api.github.com/repos/pytorch/pytorch/contents/.circleci" + url = "https://api.github.com/repos/pytorch/pytorch/contents/.ci" response = requests.get(url, headers=REQUEST_HEADERS) - for file in response.json(): - if file["name"] == "docker": - print(file["sha"]) + docker_sha = None + for finfo in response.json(): + if finfo["name"] == "docker": + docker_sha = finfo["sha"] + break + if docker_sha is None: + raise RuntimeError("Can't find sha sum of docker folder") + print(docker_sha)