@@ -173,7 +173,7 @@ pytorch_tutorial_build_worker_defaults: &pytorch_tutorial_build_worker_defaults
173
173
environment :
174
174
DOCKER_IMAGE : " 308535385114.dkr.ecr.us-east-1.amazonaws.com/pytorch/pytorch-linux-xenial-cuda10.2-cudnn7-py3-gcc7"
175
175
CUDA_VERSION : " 9"
176
- resource_class : gpu.medium
176
+ resource_class : gpu.nvidia.small
177
177
<< : *pytorch_tutorial_build_defaults
178
178
179
179
pytorch_tutorial_build_manager_defaults : &pytorch_tutorial_build_manager_defaults
@@ -223,6 +223,7 @@ jobs:
223
223
224
224
pytorch_tutorial_pr_build_worker_3 :
225
225
<< : *pytorch_tutorial_build_worker_defaults
226
+ resource_class : gpu.nvidia.small.multi
226
227
227
228
pytorch_tutorial_pr_build_worker_4 :
228
229
<< : *pytorch_tutorial_build_worker_defaults
@@ -253,12 +254,14 @@ jobs:
253
254
254
255
pytorch_tutorial_pr_build_worker_13 :
255
256
<< : *pytorch_tutorial_build_worker_defaults
257
+ resource_class : gpu.nvidia.small.multi
256
258
257
259
pytorch_tutorial_pr_build_worker_14 :
258
260
<< : *pytorch_tutorial_build_worker_defaults
259
261
260
262
pytorch_tutorial_pr_build_worker_15 :
261
263
<< : *pytorch_tutorial_build_worker_defaults
264
+ resource_class : gpu.nvidia.medium
262
265
263
266
pytorch_tutorial_pr_build_worker_16 :
264
267
<< : *pytorch_tutorial_build_worker_defaults
0 commit comments