diff --git a/.github/scripts/validate_binaries.sh b/.github/scripts/validate_binaries.sh index 97d94ffc2..f1463ccb0 100755 --- a/.github/scripts/validate_binaries.sh +++ b/.github/scripts/validate_binaries.sh @@ -16,7 +16,6 @@ else eval ${PIP_INSTALLATION} eval ${PYTHON_PATH}/python ./test/smoke_test/smoke_test.py --package torchonly else - # Special case Pypi installation package, only applicable to linux nightly CUDA 11.7 builds, wheel package if [[ ${TARGET_OS} == 'linux' && ${MATRIX_CHANNEL} == 'nightly' && ${MATRIX_GPU_ARCH_VERSION} == '11.7' && ${MATRIX_PACKAGE_TYPE} == 'manywheel' ]]; then conda create -yp ${ENV_NAME}_pypi python=${MATRIX_PYTHON_VERSION} numpy @@ -40,6 +39,10 @@ else ${PWD}/check_binary.sh fi + + if [[ ${TARGET_OS} == 'windows' && ${MATRIX_CHANNEL} == 'nightly' && ${MATRIX_GPU_ARCH_VERSION} == '11.7' ]]; then + conda install -y libnvjpeg-dev -c nvidia + fi python ./test/smoke_test/smoke_test.py fi fi