Skip to content

Commit 6b325fc

Browse files
authored
[CI] Update CUDA to 11.7 (#2193)
As 11.6 builds has been disabled by pytorch/pytorch#93406 Also, update `.jenkins/get_docker_tag.py` to search for docker folder in `.ci` directory (that was moved there by pytorch/pytorch#93104 ) and raise an error if folder can not be found
1 parent b6e9dcb commit 6b325fc

File tree

3 files changed

+13
-8
lines changed

3 files changed

+13
-8
lines changed

.circleci/config.yml

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -142,14 +142,14 @@ pytorch_tutorial_build_defaults: &pytorch_tutorial_build_defaults
142142
143143
pytorch_tutorial_build_worker_defaults: &pytorch_tutorial_build_worker_defaults
144144
environment:
145-
DOCKER_IMAGE: "308535385114.dkr.ecr.us-east-1.amazonaws.com/pytorch/pytorch-linux-bionic-cuda11.6-cudnn8-py3-gcc7"
145+
DOCKER_IMAGE: "308535385114.dkr.ecr.us-east-1.amazonaws.com/pytorch/pytorch-linux-bionic-cuda11.7-cudnn8-py3-gcc7"
146146
CUDA_VERSION: "9"
147147
resource_class: gpu.nvidia.small
148148
<<: *pytorch_tutorial_build_defaults
149149

150150
pytorch_tutorial_build_manager_defaults: &pytorch_tutorial_build_manager_defaults
151151
environment:
152-
DOCKER_IMAGE: "308535385114.dkr.ecr.us-east-1.amazonaws.com/pytorch/pytorch-linux-bionic-cuda11.6-cudnn8-py3-gcc7"
152+
DOCKER_IMAGE: "308535385114.dkr.ecr.us-east-1.amazonaws.com/pytorch/pytorch-linux-bionic-cuda11.7-cudnn8-py3-gcc7"
153153
resource_class: medium
154154
<<: *pytorch_tutorial_build_defaults
155155

.circleci/config.yml.in

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -142,14 +142,14 @@ pytorch_tutorial_build_defaults: &pytorch_tutorial_build_defaults
142142

143143
pytorch_tutorial_build_worker_defaults: &pytorch_tutorial_build_worker_defaults
144144
environment:
145-
DOCKER_IMAGE: "308535385114.dkr.ecr.us-east-1.amazonaws.com/pytorch/pytorch-linux-bionic-cuda11.6-cudnn8-py3-gcc7"
145+
DOCKER_IMAGE: "308535385114.dkr.ecr.us-east-1.amazonaws.com/pytorch/pytorch-linux-bionic-cuda11.7-cudnn8-py3-gcc7"
146146
CUDA_VERSION: "9"
147147
resource_class: gpu.nvidia.small
148148
<<: *pytorch_tutorial_build_defaults
149149

150150
pytorch_tutorial_build_manager_defaults: &pytorch_tutorial_build_manager_defaults
151151
environment:
152-
DOCKER_IMAGE: "308535385114.dkr.ecr.us-east-1.amazonaws.com/pytorch/pytorch-linux-bionic-cuda11.6-cudnn8-py3-gcc7"
152+
DOCKER_IMAGE: "308535385114.dkr.ecr.us-east-1.amazonaws.com/pytorch/pytorch-linux-bionic-cuda11.7-cudnn8-py3-gcc7"
153153
resource_class: medium
154154
<<: *pytorch_tutorial_build_defaults
155155
{% raw %}

.jenkins/get_docker_tag.py

Lines changed: 9 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -5,9 +5,14 @@
55
}
66

77
if __name__ == "__main__":
8-
url = "https://api.github.com/repos/pytorch/pytorch/contents/.circleci"
8+
url = "https://api.github.com/repos/pytorch/pytorch/contents/.ci"
99

1010
response = requests.get(url, headers=REQUEST_HEADERS)
11-
for file in response.json():
12-
if file["name"] == "docker":
13-
print(file["sha"])
11+
docker_sha = None
12+
for finfo in response.json():
13+
if finfo["name"] == "docker":
14+
docker_sha = finfo["sha"]
15+
break
16+
if docker_sha is None:
17+
raise RuntimeError("Can't find sha sum of docker folder")
18+
print(docker_sha)

0 commit comments

Comments
 (0)