fix issue with GPU being available but incompatible

README.md

@@ -142,6 +142,7 @@ is closed then to get back into the environment `cd transcribe_anything` and exe
   * All tests are run by `tox`, simply go to the project directory root and run it.
 
 # Versions
+  * 2.3.9: Fallback to `cpu` device if `gpu` device is not compatible.
   * 2.3.8: Fix --models arg which
   * 2.3.7: Critical fix: fixes dependency breakage with open-ai. Fixes windows use of embedded tool.
   * 2.3.6: Fixes typo in readme for installation instructions.

setup.py

@@ -16,7 +16,7 @@
 EMAIL = "dont@email.me"
 AUTHOR = "Zach Vorhies"
 REQUIRES_PYTHON = ">=3.10.0"
-VERSION = "2.3.8"
+VERSION = "2.3.9"
 
 # The text of the README file
 with open(os.path.join(HERE, "README.md"), encoding="utf-8", mode="r") as fd:

transcribe_anything/util.py

@@ -14,7 +14,7 @@
 def get_computing_device() -> str:
     """Get the computing device."""
     try:
-        if torch.cuda.is_available():
+        if torch.cuda.is_available() and torch.cuda._check_capability():  # pylint: disable=protected-access
             return "cuda"
         return "cpu"
     except ImportError: