diff --git a/.github/scripts/validate_binaries.sh b/.github/scripts/validate_binaries.sh index 357b91621..6ce3dd70b 100755 --- a/.github/scripts/validate_binaries.sh +++ b/.github/scripts/validate_binaries.sh @@ -20,7 +20,7 @@ else # Special case Pypi installation package, only applicable to linux nightly CUDA 11.7 builds, wheel package - if [[ ${TARGET_OS} == 'linux' && ${MATRIX_GPU_ARCH_VERSION} == '11.7' && ${MATRIX_PACKAGE_TYPE} == 'manywheel' ]]; then + if [[ ${TARGET_OS} == 'linux' && ${MATRIX_GPU_ARCH_VERSION} == '11.7' && ${MATRIX_PACKAGE_TYPE} == 'manywheel' && ${MATRIX_CHANNEL} != 'nightly' ]]; then conda create -yp ${ENV_NAME}_pypi python=${MATRIX_PYTHON_VERSION} numpy ffmpeg INSTALLATION_PYPI=${MATRIX_INSTALLATION/"cu117"/"cu117_pypi_cudnn"} INSTALLATION_PYPI=${INSTALLATION_PYPI/"torchvision torchaudio"/""} @@ -35,7 +35,6 @@ else conda create -y -n ${ENV_NAME} python=${MATRIX_PYTHON_VERSION} numpy ffmpeg conda activate ${ENV_NAME} INSTALLATION=${MATRIX_INSTALLATION/"conda install"/"conda install -y"} - INSTALLATION=${INSTALLATION/"extra-index-url"/"index-url"} eval $INSTALLATION if [[ ${TARGET_OS} == 'linux' ]]; then diff --git a/test/smoke_test/smoke_test.py b/test/smoke_test/smoke_test.py index 4e9eaaf31..604fa7d44 100644 --- a/test/smoke_test/smoke_test.py +++ b/test/smoke_test/smoke_test.py @@ -36,7 +36,7 @@ def check_version(package: str) -> None: # only makes sense to check nightly package where dates are known if channel == "nightly": - check_nightly_binaries_date(options.package) + check_nightly_binaries_date(package) else: if not torch.__version__.startswith(stable_version): raise RuntimeError(