File tree Expand file tree Collapse file tree 3 files changed +13
-8
lines changed Expand file tree Collapse file tree 3 files changed +13
-8
lines changed Original file line number Diff line number Diff line change @@ -142,14 +142,14 @@ pytorch_tutorial_build_defaults: &pytorch_tutorial_build_defaults
142
142
143
143
pytorch_tutorial_build_worker_defaults : &pytorch_tutorial_build_worker_defaults
144
144
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"
146
146
CUDA_VERSION : " 9"
147
147
resource_class : gpu.nvidia.small
148
148
<< : *pytorch_tutorial_build_defaults
149
149
150
150
pytorch_tutorial_build_manager_defaults : &pytorch_tutorial_build_manager_defaults
151
151
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"
153
153
resource_class : medium
154
154
<< : *pytorch_tutorial_build_defaults
155
155
Original file line number Diff line number Diff line change @@ -142,14 +142,14 @@ pytorch_tutorial_build_defaults: &pytorch_tutorial_build_defaults
142
142
143
143
pytorch_tutorial_build_worker_defaults: &pytorch_tutorial_build_worker_defaults
144
144
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"
146
146
CUDA_VERSION: "9"
147
147
resource_class: gpu.nvidia.small
148
148
<<: *pytorch_tutorial_build_defaults
149
149
150
150
pytorch_tutorial_build_manager_defaults: &pytorch_tutorial_build_manager_defaults
151
151
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"
153
153
resource_class: medium
154
154
<<: *pytorch_tutorial_build_defaults
155
155
{% raw %}
Original file line number Diff line number Diff line change 5
5
}
6
6
7
7
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 "
9
9
10
10
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 )
You can’t perform that action at this time.
0 commit comments