File tree Expand file tree Collapse file tree 2 files changed +4
-4
lines changed Expand file tree Collapse file tree 2 files changed +4
-4
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 %}
You can’t perform that action at this time.
0 commit comments