mirror of
https://github.com/RVC-Boss/GPT-SoVITS.git
synced 2025-04-05 19:41:56 +08:00
config.py :
现在在检查半精度计算前会先检查CUDA webui.py : 优化users.py的写入
This commit is contained in:
parent
f9387e0af8
commit
a0af8c32de
@ -29,6 +29,7 @@ webui_port_subfix = 9871
|
||||
|
||||
api_port = 9880
|
||||
|
||||
if infer_device == "cuda":
|
||||
gpu_name = torch.cuda.get_device_name(0)
|
||||
if (
|
||||
("16" in gpu_name and "V100" not in gpu_name.upper())
|
||||
|
10
webui.py
10
webui.py
@ -25,13 +25,21 @@ for path in site.getsitepackages():
|
||||
if(site_packages_roots==[]):site_packages_roots=["%s/runtime/Lib/site-packages" % now_dir]
|
||||
#os.environ["OPENBLAS_NUM_THREADS"] = "4"
|
||||
os.environ["no_proxy"] = "localhost, 127.0.0.1, ::1"
|
||||
os.environ["all_proxy"] = ""
|
||||
|
||||
for site_packages_root in site_packages_roots:
|
||||
if os.path.exists(site_packages_root):
|
||||
with open("%s/users.pth" % (site_packages_root), "w") as f:
|
||||
print(site_packages_roots)
|
||||
try:
|
||||
with open("%s/users.pth" % (site_packages_root), "w+") as f:
|
||||
f.write(
|
||||
"%s\n%s/tools\n%s/tools/damo_asr\n%s/GPT_SoVITS\n%s/tools/uvr5"
|
||||
% (now_dir, now_dir, now_dir, now_dir, now_dir)
|
||||
)
|
||||
break
|
||||
except PermissionError:
|
||||
pass
|
||||
|
||||
from tools import my_utils
|
||||
import traceback
|
||||
import shutil
|
||||
|
Loading…
x
Reference in New Issue
Block a user