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 34
34
fail-fast : false
35
35
runs-on : ${{ matrix.runner }}
36
36
env :
37
- DOCKER_IMAGE : " 308535385114.dkr.ecr.us-east-1.amazonaws.com/pytorch/pytorch-linux-bionic -cuda12.1-cudnn8-py3-gcc9"
37
+ DOCKER_IMAGE : " 308535385114.dkr.ecr.us-east-1.amazonaws.com/pytorch/pytorch-linux-focal -cuda12.1-cudnn8-py3-gcc9"
38
38
CUDA_VERSION : " 9"
39
39
steps :
40
40
- name : Setup SSH (Click me for login details)
@@ -115,9 +115,9 @@ jobs:
115
115
name : pytorch_tutorial_build_manager
116
116
needs : worker
117
117
runs-on : [self-hosted, linux.2xlarge]
118
- environment : pytorchbot-env
118
+ environment : ${{ github.ref == 'refs/heads/main' && ' pytorchbot-env' || '' }}
119
119
env :
120
- DOCKER_IMAGE : " 308535385114.dkr.ecr.us-east-1.amazonaws.com/pytorch/pytorch-linux-bionic -cuda12.1-cudnn8-py3-gcc9"
120
+ DOCKER_IMAGE : " 308535385114.dkr.ecr.us-east-1.amazonaws.com/pytorch/pytorch-linux-focal -cuda12.1-cudnn8-py3-gcc9"
121
121
CUDA_VERSION : " 9"
122
122
steps :
123
123
- name : Setup SSH (Click me for login details)
You can’t perform that action at this time.
0 commit comments