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
142142
143143pytorch_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
150150pytorch_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
Original file line number Diff line number Diff line change @@ -142,14 +142,14 @@ pytorch_tutorial_build_defaults: &pytorch_tutorial_build_defaults
142142
143143pytorch_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
150150pytorch_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 %}
Original file line number Diff line number Diff line change 55}
66
77if __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 )
You can’t perform that action at this time.
0 commit comments