Add pypi cudnn package to tests (pytorch#1168)

* Add pypi cudnn package to tests

* Fix pypi installation check

* Fix pypi instructions setting

Author: atalman

.github/actions/validate-binary/action.yml

@@ -13,6 +13,10 @@ inputs:
     description: 'Installation instructions'
     required: true
     default: ''
+  installation_pypi:
+    description: 'Installation instructions for pypi'
+    required: false
+    default: ''
   python_version:
     description: 'Python version'
     required: true
@@ -60,13 +64,16 @@ runs:
         GPU_ARCH_VER: ${{ inputs.gpu_arch_ver }}
         GPU_ARCH_TYPE: ${{ inputs.gpu_arch_type }}
         INSTALLATION: ${{ inputs.installation }}
+        INSTALLATION_PYPI: ${{ inputs.installation_pypi }}
         ENV_NAME: conda-env-${{ github.run_id }}
         DESIRED_PYTHON: ${{ inputs.python_version }}
         DESIRED_CUDA: ${{ inputs.desired_cuda }}
         DESIRED_DEVTOOLSET: ${{ inputs.dev_toolset }}
         PACKAGE_TYPE: ${{ inputs.package_type }}
+        TARGET_OS: ${{ inputs.target_os }}
       run: |
 
+        # Special case Python 3.11 wheels
         if [ $DESIRED_PYTHON == '3.11' ]; then
           set -ex
           export CPYTHON_VERSIONS=3.11.0
@@ -84,6 +91,15 @@ runs:
           eval ${PYTHON_PATH}/python ./test/smoke_test/smoke_test.py --package torchonly
         else
           set -ex
+
+          # Special case Pypi installation instructions
+          if [ ! -z "${INSTALLATION_PYPI}" ]; then
+            conda create -yp ${ENV_NAME}_pypi python=${{ inputs.python_version }} numpy
+            conda run -p ${ENV_NAME}_pypi $INSTALLATION_PYPI
+            conda run -p ${ENV_NAME}_pypi python3 ./test/smoke_test/smoke_test.py --package torchonly
+            conda env remove -p ${ENV_NAME}_pypi
+          fi
+
           conda create -yp ${ENV_NAME} python=${{ inputs.python_version }} numpy
           conda run -p ${ENV_NAME} $INSTALLATION
           conda run -p ${ENV_NAME} python3  ./test/smoke_test/smoke_test.py

.github/workflows/validate-linux-binaries.yml

@@ -76,6 +76,7 @@ jobs:
           gpu_arch_type: ${{ matrix.gpu_arch_type }}
           gpu_arch_ver: ${{ matrix.gpu_arch_version }}
           installation: ${{ matrix.installation }}
+          installation_pypi: ${{ matrix.installation_pypi }}
           python_version: ${{ matrix.python_version }}
           desired_cuda: ${{ matrix.desired_cuda }}
           dev_toolset: ''