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 3434 fail-fast : false
3535 runs-on : ${{ matrix.runner }}
3636 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"
3838 CUDA_VERSION : " 9"
3939 steps :
4040 - name : Setup SSH (Click me for login details)
@@ -115,9 +115,9 @@ jobs:
115115 name : pytorch_tutorial_build_manager
116116 needs : worker
117117 runs-on : [self-hosted, linux.2xlarge]
118- environment : pytorchbot-env
118+ environment : ${{ github.ref == 'refs/heads/main' && ' pytorchbot-env' || '' }}
119119 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"
121121 CUDA_VERSION : " 9"
122122 steps :
123123 - name : Setup SSH (Click me for login details)
You can’t perform that action at this time.
0 commit comments