33
33
- { shard: 15, num_shards: 15, runner: "linux.4xlarge.nvidia.gpu" }
34
34
fail-fast : false
35
35
runs-on : ${{ matrix.runner }}
36
- env :
37
- DOCKER_IMAGE : " 308535385114.dkr.ecr.us-east-1.amazonaws.com/pytorch/pytorch-linux-focal-cuda12.1-cudnn8-py3-gcc9"
38
- CUDA_VERSION : " 9"
39
36
steps :
40
37
- name : Setup SSH (Click me for login details)
41
38
uses : pytorch/test-infra/.github/actions/setup-ssh@main
54
51
- name : Install nvidia driver, nvidia-docker runtime, set GPU_FLAG
55
52
uses : pytorch/test-infra/.github/actions/setup-nvidia@main
56
53
57
- - name : Build docker image
54
+ - name : Calculate/build docker image
58
55
id : calculate-docker-image
59
56
uses : pytorch/test-infra/.github/actions/calculate-docker-image@main
60
57
with :
@@ -104,14 +101,10 @@ jobs:
104
101
if : always()
105
102
106
103
manager :
107
- if : false
108
104
name : pytorch_tutorial_build_manager
109
105
needs : worker
110
106
runs-on : [self-hosted, linux.2xlarge]
111
107
environment : ${{ github.ref == 'refs/heads/main' && 'pytorchbot-env' || '' }}
112
- env :
113
- DOCKER_IMAGE : " 308535385114.dkr.ecr.us-east-1.amazonaws.com/pytorch/pytorch-linux-focal-cuda12.1-cudnn8-py3-gcc9"
114
- CUDA_VERSION : " 9"
115
108
steps :
116
109
- name : Setup SSH (Click me for login details)
117
110
uses : pytorch/test-infra/.github/actions/setup-ssh@main
@@ -127,7 +120,7 @@ jobs:
127
120
- name : Setup Linux
128
121
uses : pytorch/pytorch/.github/actions/setup-linux@main
129
122
130
- - name : Build docker image
123
+ - name : Calculate/build docker image
131
124
id : calculate-docker-image
132
125
uses : pytorch/test-infra/.github/actions/calculate-docker-image@main
133
126
with :
0 commit comments