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
141141
142142pytorch_tutorial_build_worker_defaults : &pytorch_tutorial_build_worker_defaults
143143 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"
145145 CUDA_VERSION : " 9"
146146 resource_class : gpu.nvidia.small
147147 << : *pytorch_tutorial_build_defaults
148148
149149pytorch_tutorial_build_manager_defaults : &pytorch_tutorial_build_manager_defaults
150150 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"
152152 resource_class : medium
153153 << : *pytorch_tutorial_build_defaults
154154
Original file line number Diff line number Diff line change @@ -141,14 +141,14 @@ pytorch_tutorial_build_defaults: &pytorch_tutorial_build_defaults
141141
142142pytorch_tutorial_build_worker_defaults: &pytorch_tutorial_build_worker_defaults
143143 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"
145145 CUDA_VERSION: "9"
146146 resource_class: gpu.nvidia.small
147147 <<: *pytorch_tutorial_build_defaults
148148
149149pytorch_tutorial_build_manager_defaults: &pytorch_tutorial_build_manager_defaults
150150 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"
152152 resource_class: medium
153153 <<: *pytorch_tutorial_build_defaults
154154{% raw %}
You can’t perform that action at this time.
0 commit comments