test

Author: lanluo-nvidia

.github/workflows/release-wheel-windows.yml

@@ -127,7 +127,10 @@ jobs:
           # shellcheck disable=SC2086
           ${CONDA_RUN} ${PIP_INSTALL_TORCH}
           set -x
+          ${CONDA_RUN} python -c "import torch; print(torch.cuda.is_available())"
+          ${CONDA_RUN} python -c "import torch; print(torch.cuda.get_device_name(0))"
           ${CONDA_RUN} python -c "import torch; print(torch.cuda.get_device_properties(0))"
+          nvidia-smi
 
       # - name: Run Pre-Script with Caching
       #   if: ${{ inputs.pre-script != '' }}