Skip to content

Commit fbaa01d

Browse files
author
Will Feng
committed
try something different
1 parent def5653 commit fbaa01d

File tree

1 file changed

+3
-3
lines changed

1 file changed

+3
-3
lines changed

.circleci/config.yml

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -162,14 +162,14 @@ pytorch_tutorial_build_defaults: &pytorch_tutorial_build_defaults
162162
163163
pytorch_tutorial_build_worker_defaults: &pytorch_tutorial_build_worker_defaults
164164
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"
167167
resource_class: gpu.medium
168168
<<: *pytorch_tutorial_build_defaults
169169

170170
pytorch_tutorial_build_manager_defaults: &pytorch_tutorial_build_manager_defaults
171171
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"
173173
resource_class: medium
174174
<<: *pytorch_tutorial_build_defaults
175175

0 commit comments

Comments
 (0)