Fix cuda version check

Fix Audio and Vision version check

Add check binary to libtorch

test

test

testing

testing

testing

Testing

Testing

testing

Author: atalman

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

@@ -58,22 +58,19 @@ jobs:
         export CUDA_VER="${{ matrix.desired_cuda }}"
         export DESIRED_PYTHON="${{ matrix.python_version }}"
         export DESIRED_CUDA="${{ matrix.desired_cuda }}"
-        export DESIRED_DEVTOOLSET="${{ matrix.dev_toolset }}"
+        export DESIRED_DEVTOOLSET="${{ matrix.devtoolset }}"
         export PACKAGE_TYPE="${{ matrix.package_type }}"
         export TARGET_OS="linux"
+        conda create -y -n ${ENV_NAME} python=${{ matrix.python_version }} numpy pillow
+        conda activate ${ENV_NAME}
+        export CONDA_LIBRARY_PATH="$(dirname $(which python))/../lib"
+        export LD_LIBRARY_PATH=$CONDA_LIBRARY_PATH:$LD_LIBRARY_PATH
 
         if [[ ${{ matrix.package_type }} == "libtorch" ]]; then
           curl ${{ matrix.installation }} -o libtorch.zip
           unzip libtorch.zip
         else
-          conda create -y -n ${ENV_NAME} python=${{ matrix.python_version }} numpy pillow
-          conda activate ${ENV_NAME}
           eval $INSTALLATION
-          export CONDA_LIBRARY_PATH="$(dirname $(which python))/../lib"
-          export LD_LIBRARY_PATH=$CONDA_LIBRARY_PATH:$LD_LIBRARY_PATH
-
-          python --version
           python  ./test/smoke_test/smoke_test.py
-
           ${PWD}/check_binary.sh
         fi

test/smoke_test/smoke_test.py

@@ -42,6 +42,18 @@ def check_nightly_binaries_date(package: str) -> None:
                 f"Expected torchaudio, torchvision to be less then {NIGHTLY_ALLOWED_DELTA} days. But they are from {date_ta_str}, {date_tv_str} respectively"
             )
 
+def check_cuda_version(version: str, dlibary: str):
+    version = torch.ops.torchaudio.cuda_version()
+    if version is not None and torch.version.cuda is not None:
+        version_str = str(version)
+        ta_version = f"{version_str[:-3]}.{version_str[-2]}"
+        t_version = torch.version.cuda.split(".")
+        t_version = f"{t_version[0]}.{t_version[1]}"
+        if ta_version != t_version:
+            raise RuntimeError(
+                "Detected that PyTorch and {dlibary} were compiled with different CUDA versions. "
+                f"PyTorch has CUDA version {t_version} whereas {dlibary} has CUDA version {ta_version}. "
+            )
 
 def smoke_test_cuda(package: str) -> None:
     if not torch.cuda.is_available() and is_cuda_system:
@@ -56,20 +68,14 @@ def smoke_test_cuda(package: str) -> None:
         print(f"torch cudnn: {torch.backends.cudnn.version()}")
         print(f"cuDNN enabled? {torch.backends.cudnn.enabled}")
 
-    if(package == 'all'):
+    if(package == 'all' and is_cuda_system):
         import torchaudio
         import torchvision
-
         print(f"torchvision cuda: {torch.ops.torchvision._cuda_version()}")
         print(f"torchaudio cuda: {torch.ops.torchaudio.cuda_version()}")
-        if (
-            gpu_arch_ver != torch.ops.torchvision._cuda_version() or
-            gpu_arch_ver != torch.ops.torchaudio.cuda_version()
-            ):
-            raise RuntimeError(
-                    f"Wrong CUDA version. Vision: {torch.ops.torchvision._cuda_version()} \
-                    Audio: {ttorch.ops.torchaudio.cuda_version()} Expected: {gpu_arch_ver}"
-                )
+        check_cuda_version(torch.ops.torchvision._cuda_version(), "TorchVision")
+        check_cuda_version(torch.ops.torchaudio.cuda_version(), "TorchAudio")
+
 
 def smoke_test_conv2d() -> None:
     import torch.nn as nn