File tree Expand file tree Collapse file tree 1 file changed +3
-3
lines changed Expand file tree Collapse file tree 1 file changed +3
-3
lines changed Original file line number Diff line number Diff line change @@ -162,14 +162,14 @@ pytorch_tutorial_build_defaults: &pytorch_tutorial_build_defaults
162
162
163
163
pytorch_tutorial_build_worker_defaults : &pytorch_tutorial_build_worker_defaults
164
164
environment :
165
- DOCKER_IMAGE : " 308535385114.dkr.ecr.us-east-1.amazonaws.com/pytorch/pytorch-linux-xenial-cuda10 -cudnn7-py3-gcc7 "
166
- CUDA_VERSION : " 10 "
165
+ DOCKER_IMAGE : " 308535385114.dkr.ecr.us-east-1.amazonaws.com/pytorch/pytorch-linux-xenial-cuda9 -cudnn7-py3"
166
+ CUDA_VERSION : " 9 "
167
167
resource_class : gpu.medium
168
168
<< : *pytorch_tutorial_build_defaults
169
169
170
170
pytorch_tutorial_build_manager_defaults : &pytorch_tutorial_build_manager_defaults
171
171
environment :
172
- DOCKER_IMAGE : " 308535385114.dkr.ecr.us-east-1.amazonaws.com/pytorch/pytorch-linux-xenial-cuda10 -cudnn7-py3-gcc7 "
172
+ DOCKER_IMAGE : " 308535385114.dkr.ecr.us-east-1.amazonaws.com/pytorch/pytorch-linux-xenial-cuda9 -cudnn7-py3"
173
173
resource_class : medium
174
174
<< : *pytorch_tutorial_build_defaults
175
175
You can’t perform that action at this time.
0 commit comments