diff --git a/.github/workflows/validate-binaries.yml b/.github/workflows/validate-binaries.yml index 248857214..44f5e6e4d 100644 --- a/.github/workflows/validate-binaries.yml +++ b/.github/workflows/validate-binaries.yml @@ -38,4 +38,5 @@ jobs: channel: ${{ inputs.channel }} repository: "pytorch/torchrec" smoke_test: "source ./.github/scripts/validate_binaries.sh" + post_script: .github/scripts/install_libs.sh with_cuda: enable diff --git a/.github/workflows/validate-nightly-binaries.yml b/.github/workflows/validate-nightly-binaries.yml index 6d6369495..c0478b35c 100644 --- a/.github/workflows/validate-nightly-binaries.yml +++ b/.github/workflows/validate-nightly-binaries.yml @@ -24,3 +24,4 @@ jobs: uses: ./.github/workflows/validate-binaries.yml with: channel: nightly + post_script: .github/scripts/install_libs.sh