SD运行 webui-user.bat 报错Torch is not able to use GPU的报错原因及解决办法

本文介绍了如何解决深度学习项目中遇到的错误,重点在于更新NVIDIA驱动、安装对应CUDA版本、cuDNN的配置,以及使用特定版本的torch、torchvision和torchaudio。作者提供了详细步骤,包括替换launch.py或launch_utils.py中的pip安装命令,以避免显卡问题并确保兼容性。

摘要生成于 C知道 ,由 DeepSeek-R1 满血版支持, 前往体验 >

        首先网上大部分博主讲解的在 launch.py 或者 webui-user.bat加一个参数 --precision full --no-half --skip-torch-cuda-test ,加上之后确实不报错了,但是你的显卡也不工作了,CPU进入了AI算法时代,这个方法治标不治本,如果有显卡且拥有4G显存以上的同学,不妨试试我这种办法,虽然不全面,但至少解决了我的问题,大家也可以作为参考。

        进行检错前需要搞清楚几个点
1.你的NVIDIA驱动是否是最新版本,驱动太旧也会报错,但概率不大,可以先往下看。

2.安装对应版本的CUDA,根据我的经验来看,系统推荐安装的版本一直是官网没有的,且降级安装不一定有对应的torch和xformers插件版本,后续会导致其他报错,不如稳一点,用目前稳定的推荐版本。cuda_12.1.0_531.14_windows.exeCUDA Toolkit 12.1 Downloads | NVIDIA Developer

评论 2
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值