zlei6/ComfyUI
1
0
Fork 0
mirror of https://github.com/Comfy-Org/ComfyUI.git synced 2026-09-08 02:08:37 +08:00
Code Issues Packages Projects Releases Wiki Activity

Add warning about forcing single GPU mode on Windows (#16055)

Browse Source
This commit is contained in:
comfyanonymous
2026-09-02 19:48:06 -07:00
committed by GitHub
parent 1af040bf02
commit 345c919049
1 changed files with 1 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

1
main.py
View File

@@ -50,6 +50,7 @@ if __name__ == "__main__":
and os.environ.get("CUDA_VISIBLE_DEVICES") is None
):
os.environ["CUDA_VISIBLE_DEVICES"] = "0"
logging.warning("On windows we are currently forcing single GPU mode in ComfyUI due to a Nvidia related issue, if you want to disable this use: --cuda-device all")
faulthandler.enable(file=sys.stderr, all_threads=args.debug_hang)
if __name__ == "__main__" and args.debug_hang: