blackteay/GPT-SoVITS
1
0
Fork 0
mirror of https://github.com/RVC-Boss/GPT-SoVITS.git synced 2025-08-17 15:09:55 +08:00
Code Issues Packages Projects Releases Wiki Activity

Fix Path Problems

Browse Source
This commit is contained in:
XXXXRT666 2025-05-12 10:03:43 +01:00
parent d5e3dbf09f
commit 0d30e2d187
1 changed files with 6 additions and 10 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

16
tools/my_utils.py
View File

@ -147,8 +147,7 @@ def load_cudnn():
if torch_lib_dir.exists():
os.add_dll_directory(str(torch_lib_dir))
print(f"[INFO] Added DLL directory: {torch_lib_dir}")
pattern = str(torch_lib_dir / "cudnn_cnn*.dll")
matching_files = sorted(glob.glob(pattern))
matching_files = sorted(torch_lib_dir.glob("cudnn_cnn*.dll"))
if not matching_files:
print(f"[ERROR] No cudnn_cnn*.dll found in {torch_lib_dir}")
return
@ -170,15 +169,14 @@ def load_cudnn():
print(f"[ERROR] cudnn dir not found: {cudnn_dir}")
return
pattern = str(cudnn_dir / "libcudnn_cnn*.so*")
matching_files = sorted(glob.glob(pattern))
matching_files = sorted(cudnn_dir.glob("libcudnn_cnn*.so*"))
if not matching_files:
print(f"[ERROR] No libcudnn_cnn*.so* found in {cudnn_dir}")
return
for so_path in matching_files:
try:
ctypes.CDLL(so_path, mode=ctypes.RTLD_GLOBAL)
ctypes.CDLL(so_path, mode=ctypes.RTLD_GLOBAL) # type: ignore
print(f"[INFO] Loaded: {so_path}")
except OSError as e:
print(f"[WARNING] Failed to load {so_path}: {e}")
@ -196,8 +194,7 @@ def load_nvrtc():
if torch_lib_dir.exists():
os.add_dll_directory(str(torch_lib_dir))
print(f"[INFO] Added DLL directory: {torch_lib_dir}")
pattern = str(torch_lib_dir / "nvrtc*.dll")
matching_files = sorted(glob.glob(pattern))
matching_files = sorted(torch_lib_dir.glob("nvrtc*.dll"))
if not matching_files:
print(f"[ERROR] No nvrtc*.dll found in {torch_lib_dir}")
return
@ -219,15 +216,14 @@ def load_nvrtc():
print(f"[ERROR] nvrtc dir not found: {nvrtc_dir}")
return
pattern = str(nvrtc_dir / "libnvrtc*.so*")
matching_files = sorted(glob.glob(pattern))
matching_files = sorted(nvrtc_dir.glob("libnvrtc*.so*"))
if not matching_files:
print(f"[ERROR] No libnvrtc*.so* found in {nvrtc_dir}")
return
for so_path in matching_files:
try:
ctypes.CDLL(so_path, mode=ctypes.RTLD_GLOBAL)
ctypes.CDLL(so_path, mode=ctypes.RTLD_GLOBAL) # type: ignore
print(f"[INFO] Loaded: {so_path}")
except OSError as e:
print(f"[WARNING] Failed to load {so_path}: {e}")