Skip to content

Commit e7c86fd

Browse files
authored
Update docker image and add if statement for deployment env (#2527)
* update docker image * also change env because the manager won't work
1 parent 26dd549 commit e7c86fd

File tree

1 file changed

+3
-3
lines changed

1 file changed

+3
-3
lines changed

.github/workflows/build-tutorials.yml

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -34,7 +34,7 @@ jobs:
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)

0 commit comments

Comments
 (0)