Skip to content

Commit 51ca1f5

Browse files
authored
Add pypi with cudnn package test (#1289)
* Add pypi with cudnn package test * Add pypi with cudnn package test * test * test * More pypi cudnn changes * test * Fix pipy smoke test * Remove debug comments
1 parent b0071ac commit 51ca1f5

File tree

1 file changed

+14
-0
lines changed

1 file changed

+14
-0
lines changed

.github/scripts/validate_binaries.sh

Lines changed: 14 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -2,6 +2,7 @@ if [[ ${MATRIX_PACKAGE_TYPE} == "libtorch" ]]; then
22
curl ${MATRIX_INSTALLATION} -o libtorch.zip
33
unzip libtorch.zip
44
else
5+
#special case for Python 3.11
56
if [ $MATRIX_PYTHON_VERSION == '3.11' ]; then
67
export CPYTHON_VERSIONS=3.11.0
78
sudo yum -y install openssl-devel libssl-dev bzip2-devel libffi-devel
@@ -15,6 +16,19 @@ else
1516
eval ${PIP_INSTALLATION}
1617
eval ${PYTHON_PATH}/python ./test/smoke_test/smoke_test.py --package torchonly
1718
else
19+
20+
# Special case Pypi installation package, only applicable to linux nightly CUDA 11.7 builds, wheel package
21+
if [[ ${TARGET_OS} == 'linux' && ${MATRIX_CHANNEL} == 'nightly' && ${MATRIX_GPU_ARCH_VERSION} == '11.7' && ${MATRIX_PACKAGE_TYPE} == 'manywheel' ]]; then
22+
conda create -yp ${ENV_NAME}_pypi python=${MATRIX_PYTHON_VERSION} numpy
23+
INSTALLATION_PYPI=${MATRIX_INSTALLATION/"cu117"/"cu117_pypi_cudnn"}
24+
INSTALLATION_PYPI=${INSTALLATION_PYPI/"torchvision torchaudio"/""}
25+
INSTALLATION_PYPI=${INSTALLATION_PYPI/"index-url"/"extra-index-url"}
26+
conda run -p ${ENV_NAME}_pypi ${INSTALLATION_PYPI}
27+
conda run -p ${ENV_NAME}_pypi python ./test/smoke_test/smoke_test.py --package torchonly
28+
conda activate base
29+
conda env remove -p ${ENV_NAME}_pypi
30+
fi
31+
1832
conda create -y -n ${ENV_NAME} python=${MATRIX_PYTHON_VERSION} numpy pillow
1933
conda activate ${ENV_NAME}
2034
INSTALLATION=${MATRIX_INSTALLATION/"conda install"/"conda install -y"}

0 commit comments

Comments
 (0)