From 99c91d20b2ab0dba48b00dfb3174cb0b85f92b52 Mon Sep 17 00:00:00 2001 From: Felipe Daragon Date: Wed, 16 Apr 2025 21:14:48 +0100 Subject: [PATCH] Supress CUDA path error in CPU mode --- README.md | 4 ++-- refacer.py | 8 ++++++-- 2 files changed, 8 insertions(+), 4 deletions(-) diff --git a/README.md b/README.md index 7a82f4d..8f32505 100644 --- a/README.md +++ b/README.md @@ -77,7 +77,7 @@ On Windows, before continuing, ensure that you have the [Visual Studio Build Too Follow these steps to install Refacer and its dependencies: - ```bash +```bash # Check if ffmpeg is available (if not, you might to download it and add it to your PATH) # Windows: download ffmpeg-git-essentials.7z from https://www.gyan.dev/ffmpeg/builds/ # Other systems: see a tutorial https://www.hostinger.com/tutorials/how-to-install-ffmpeg @@ -107,7 +107,7 @@ Follow these steps to install Refacer and its dependencies: # For CoreML only (compatible with MacOSX, requires Silicon architecture): pip install -r requirements-COREML.txt - ``` +``` For NVIDIA GPU, make sure you have both NVIDIA GPU Computing Toolkit and NVIDIA CUDNN installed. The onnxruntime-gpu version must match your version of CUDA. This example uses onnxruntime-gpu 1.21.0, which is compatible with CUDA 12.6 and CUDNN 9.4 - Refacer.py is pre-loading both libraries. Remember to update the paths if needed in refacer.py if you have different location or versions. diff --git a/refacer.py b/refacer.py index 4a2f6af..0c9b6b7 100644 --- a/refacer.py +++ b/refacer.py @@ -30,8 +30,12 @@ gc = __import__('gc') # Preload NVIDIA DLLs if Windows if sys.platform in ("win32", "win64"): if hasattr(os, "add_dll_directory"): - os.add_dll_directory(r"C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v12.6\bin") - os.add_dll_directory(r"C:\Program Files\NVIDIA\CUDNN\v9.4\bin\12.6") + try: + os.add_dll_directory(r"C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v12.6\bin") + os.add_dll_directory(r"C:\Program Files\NVIDIA\CUDNN\v9.4\bin\12.6") + except Exception as e: + print(f"[INFO] Failed to add CUDA or CUDNN DLL directory: {e}") + print("[INFO] This error can be ignored if running in CPU mode. Otherwise, make sure the paths are correct.") if hasattr(rt, "preload_dlls"): rt.preload_dlls()