防止使用webui-start时出现该错误 'Library cudart is not initialized' (Akegarasu#44)

YunCheng66 · Apr 1, 2023 · 169002e · 169002e
1 parent 32ba324
commit 169002e
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/webui.py b/webui.py
@@ -6,7 +6,7 @@
 
 from modules.options import cmd_opts
 from modules.ui import create_ui
-
+os.environ['PATH'] = os.environ.get("PATH", "") + os.pathsep + os.getcwd() + r'\.venv\Lib\site-packages\torch\lib'
 
 def ensure_output_dirs():
     folders = ["outputs/save", "outputs/markdown"]