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 %} 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)