Skip to content

Commit

Permalink
Amazon Linux 2023 validation (#6154)
Browse files Browse the repository at this point in the history
To be able to detect issues like these:
pytorch/pytorch#144433
Using script from: pytorch/pytorch#138324
  • Loading branch information
atalman authored Jan 9, 2025
1 parent 73fbf31 commit 73eea90
Showing 1 changed file with 26 additions and 0 deletions.
26 changes: 26 additions & 0 deletions .github/workflows/validate-linux-binaries.yml
Original file line number Diff line number Diff line change
Expand Up @@ -169,3 +169,29 @@ jobs:
# Validate binaries
source ../../test-infra/.github/scripts/validate_binaries.sh
linux-amazon-2023:
uses: ./.github/workflows/linux_job.yml
name: amazon-linux-2023-test
with:
runner: "linux.g5.4xlarge.nvidia.gpu"
repository: "pytorch/pytorch"
ref: main
job-name: "amazon-linux-2023-test"
docker-image: 'almalinux/9-base'
docker-build-dir: "skip-docker-build"
timeout: 180
script: |
set -ex
cd .ci/pytorch/
python3 -m ensurepip --upgrade
DWN_PYTORCH_ORG="https://download.pytorch.org/whl/nightly/cu124"
if [[ ${{ inputs.channel }} == 'test' ]]; then
DWN_PYTORCH_ORG="https://download.pytorch.org/whl/test/cu124"
elif [[ ${{ inputs.channel }} == 'release' ]]; then
DWN_PYTORCH_ORG="https://download.pytorch.org/whl/cu124"
fi
python3 -m pip install torch --index-url ${DWN_PYTORCH_ORG}
python3 -c "import torch"

0 comments on commit 73eea90

Please sign in to comment.