File tree 1 file changed +26
-0
lines changed
1 file changed +26
-0
lines changed Original file line number Diff line number Diff line change @@ -169,3 +169,29 @@ jobs:
169
169
170
170
# Validate binaries
171
171
source ../../test-infra/.github/scripts/validate_binaries.sh
172
+
173
+ linux-amazon-2023 :
174
+ uses : ./.github/workflows/linux_job.yml
175
+ name : amazon-linux-2023-test
176
+ with :
177
+ runner : " linux.g5.4xlarge.nvidia.gpu"
178
+ repository : " pytorch/pytorch"
179
+ ref : main
180
+ job-name : " amazon-linux-2023-test"
181
+ docker-image : ' almalinux/9-base'
182
+ docker-build-dir : " skip-docker-build"
183
+ timeout : 180
184
+ script : |
185
+ set -ex
186
+ cd .ci/pytorch/
187
+ python3 -m ensurepip --upgrade
188
+
189
+ DWN_PYTORCH_ORG="https://download.pytorch.org/whl/nightly/cu124"
190
+ if [[ ${{ inputs.channel }} == 'test' ]]; then
191
+ DWN_PYTORCH_ORG="https://download.pytorch.org/whl/test/cu124"
192
+ elif [[ ${{ inputs.channel }} == 'release' ]]; then
193
+ DWN_PYTORCH_ORG="https://download.pytorch.org/whl/cu124"
194
+ fi
195
+
196
+ python3 -m pip install torch --index-url ${DWN_PYTORCH_ORG}
197
+ python3 -c "import torch"
You can’t perform that action at this time.
0 commit comments