webui forge报错：AssertionError: Torch is not able to use GPU；跳过GPU检测_assertionerror: torch is not able to use gpu; add

作者：盐析白兔 | 2024-06-21 11:04:17

踩

assertionerror: torch is not able to use gpu; add --skip-torch-cuda-test to

webui forge报错：AssertionError: Torch is not able to use GPU

因为自己的torch安装在自己的conda环境里，实在不想用一个整合包又下一个torch，所以直接跳过检测。

./webui/modules/launch_utils.py 中 432行


    if not args.skip_torch_cuda_test and not check_run_python("import torch; assert torch.cuda.is_available()"):
        pass    
        '''raise RuntimeError(
            'Torch is not able to use GPU; '
            'add --skip-torch-cuda-test to COMMANDLINE_ARGS variable to disable this check'
        )'''

直接注释掉得了

声明：本文内容由网友自发贡献，不代表【wpsshop博客】立场，版权归原作者所有，本站不承担相应法律责任。如您发现有侵权的内容，请联系我们。转载请注明出处：https://www.wpsshop.cn/w/盐析白兔/article/detail/742968