Validate pypi binaries for cuda 12.4 rather then 12.1

pytorch · Oct 17, 2024 · 3f5267e · 3f5267e
1 parent bcd0972
commit 3f5267e
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/.github/workflows/validate-linux-binaries.yml b/.github/workflows/validate-linux-binaries.yml
@@ -168,7 +168,7 @@ jobs:
 
  # Special case PyPi installation package. And Install of PyPi package via poetry
  if [[ ${MATRIX_PACKAGE_TYPE} == "manywheel" && \
- ${MATRIX_GPU_ARCH_VERSION} == "12.1" && \
+ ${MATRIX_GPU_ARCH_VERSION} == "12.4" && \
  ${MATRIX_CHANNEL} == "release" && \
  ${USE_ONLY_DL_PYTORCH_ORG} == "false" ]]; then
  source ./.github/scripts/validate_pipy.sh