diff --git a/.github/scripts/validate_pipy.sh b/.github/scripts/validate_pipy.sh index 85b788cf0..d840fe019 100644 --- a/.github/scripts/validate_pipy.sh +++ b/.github/scripts/validate_pipy.sh @@ -1,12 +1,18 @@ conda create -yp ${ENV_NAME}_pypi python=${MATRIX_PYTHON_VERSION} numpy ffmpeg -if [[ ${MATRIX_CHANNEL} != "release" ]]; then +TEST_SUFFIX="" +if [[ ${TORCH_ONLY} == 'true' ]]; then + TEST_SUFFIX=" --package torchonly" conda run -p ${ENV_NAME}_pypi pip3 install --pre torch --index-url "https://download.pytorch.org/whl/${MATRIX_CHANNEL}/${MATRIX_DESIRED_CUDA}_pypi_cudnn" - conda run -p ${ENV_NAME}_pypi pip3 install --pre torchvision torchaudio --index-url "https://download.pytorch.org/whl/${MATRIX_CHANNEL}/${MATRIX_DESIRED_CUDA}" else - conda run -p ${ENV_NAME}_pypi pip3 install torch torchvision torchaudio + if [[ ${MATRIX_CHANNEL} != "release" ]]; then + conda run -p ${ENV_NAME}_pypi pip3 install --pre torch --index-url "https://download.pytorch.org/whl/${MATRIX_CHANNEL}/${MATRIX_DESIRED_CUDA}_pypi_cudnn" + conda run -p ${ENV_NAME}_pypi pip3 install --pre torchvision torchaudio --index-url "https://download.pytorch.org/whl/${MATRIX_CHANNEL}/${MATRIX_DESIRED_CUDA}" + else + conda run -p ${ENV_NAME}_pypi pip3 install torch torchvision torchaudio + fi fi -conda run -p ${ENV_NAME}_pypi python ./test/smoke_test/smoke_test.py +conda run -p ${ENV_NAME}_pypi python ./test/smoke_test/smoke_test.py ${TEST_SUFFIX} --runtime-error-check disabled conda deactivate conda env remove -p ${ENV_NAME}_pypi