diff --git a/.github/workflows/docker-release.yml b/.github/workflows/docker-release.yml index 4c3cebd3ed6..a176f24c952 100644 --- a/.github/workflows/docker-release.yml +++ b/.github/workflows/docker-release.yml @@ -81,7 +81,7 @@ jobs: # To get QEMU binaries in our PATH echo "${RUNNER_TEMP}/bin" >> "${GITHUB_PATH}" # Generate PyTorch version to use - echo "PYTORCH_VERSION=$(python3 .github/scripts/generate_pytorch_version.py)" >> "${GITHUB_ENV}" + echo "PYTORCH_VERSION=$(python3 .github/scripts/generate_pytorch_version.py --no-build-suffix)" >> "${GITHUB_ENV}" - name: Setup nightly specific variables if: ${{ github.event.ref == 'refs/heads/nightly' || startsWith(github.event.ref, 'refs/tags/ciflow/nightly/') }} run: |