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 @@ -141,14 +141,14 @@ pytorch_tutorial_build_defaults: &pytorch_tutorial_build_defaults
141
141
142
142
pytorch_tutorial_build_worker_defaults : &pytorch_tutorial_build_worker_defaults
143
143
environment :
144
- DOCKER_IMAGE : " 308535385114.dkr.ecr.us-east-1.amazonaws.com/pytorch/pytorch-linux-xenial-cuda10.2-cudnn7 -py3-gcc7"
144
+ DOCKER_IMAGE : " 308535385114.dkr.ecr.us-east-1.amazonaws.com/pytorch/pytorch-linux-bionic-cuda11.6-cudnn8 -py3-gcc7"
145
145
CUDA_VERSION : " 9"
146
146
resource_class : gpu.nvidia.small
147
147
<< : *pytorch_tutorial_build_defaults
148
148
149
149
pytorch_tutorial_build_manager_defaults : &pytorch_tutorial_build_manager_defaults
150
150
environment :
151
- DOCKER_IMAGE : " 308535385114.dkr.ecr.us-east-1.amazonaws.com/pytorch/pytorch-linux-xenial-cuda10.2-cudnn7 -py3-gcc7"
151
+ DOCKER_IMAGE : " 308535385114.dkr.ecr.us-east-1.amazonaws.com/pytorch/pytorch-linux-bionic-cuda11.6-cudnn8 -py3-gcc7"
152
152
resource_class : medium
153
153
<< : *pytorch_tutorial_build_defaults
154
154
Original file line number Diff line number Diff line change @@ -141,14 +141,14 @@ pytorch_tutorial_build_defaults: &pytorch_tutorial_build_defaults
141
141
142
142
pytorch_tutorial_build_worker_defaults: &pytorch_tutorial_build_worker_defaults
143
143
environment:
144
- DOCKER_IMAGE: "308535385114.dkr.ecr.us-east-1.amazonaws.com/pytorch/pytorch-linux-xenial-cuda10.2-cudnn7 -py3-gcc7"
144
+ DOCKER_IMAGE: "308535385114.dkr.ecr.us-east-1.amazonaws.com/pytorch/pytorch-linux-bionic-cuda11.6-cudnn8 -py3-gcc7"
145
145
CUDA_VERSION: "9"
146
146
resource_class: gpu.nvidia.small
147
147
<<: *pytorch_tutorial_build_defaults
148
148
149
149
pytorch_tutorial_build_manager_defaults: &pytorch_tutorial_build_manager_defaults
150
150
environment:
151
- DOCKER_IMAGE: "308535385114.dkr.ecr.us-east-1.amazonaws.com/pytorch/pytorch-linux-xenial-cuda10.2-cudnn7 -py3-gcc7"
151
+ DOCKER_IMAGE: "308535385114.dkr.ecr.us-east-1.amazonaws.com/pytorch/pytorch-linux-bionic-cuda11.6-cudnn8 -py3-gcc7"
152
152
resource_class: medium
153
153
<<: *pytorch_tutorial_build_defaults
154
154
{% raw %}
You can’t perform that action at this time.
0 commit comments