用stable_diffusion_webui遇到CUBLAS_STATUS_INVALID_VALUE when calling cublasGemmStridedBatchedExFix 报错

这篇具有很好参考价值的文章主要介绍了用stable_diffusion_webui遇到CUBLAS_STATUS_INVALID_VALUE when calling cublasGemmStridedBatchedExFix 报错。希望对大家有所帮助。如果存在错误或未考虑完全的地方,请大家不吝赐教,您也可以点击"举报违法"按钮提交疑问。

搭建参考教程:

1,【新修正】手把手教你在linux中部署stable-diffusion-webui N卡A卡显卡可用
2,手把手教你在linux中手动编译并安装xformers
(以上文章作者为B站up主:青空朝颜モー)

本机电脑显卡型号:2080Ti CUDA版本:11.5

问题描述

在点击生成图片时发生以下报错:

RuntimeError: CUDA error: CUBLAS_STATUS_INVALID_VALUE when calling cublasGemmStridedBatchedExFix( handle, opa, opb, m, n, k, (void*)(&falpha), a, CUDA_R_16F, lda, stridea, b, CUDA_R_16F, ldb, strideb, (void*)(&fbeta), c, CUDA_R_16F, ldc, stridec, num_batches, CUDA_R_32F, CUBLAS_GEMM_DEFAULT_TENSOR_OP)

问题分析

stable-diffusion最佳torch版本为1.13.1(时间:2/2023)。但是该版本torch支持的CUDA版本为11.6及以上(详见pytorch官网),因此会导致各种奇奇怪怪的CUDA错误产生。

但是由于升级显卡驱动与CUDA版本过于麻烦,很可能会导致电脑中原有项目出现不可使用的状况,因此慎重将已有较多项目的旧电脑升级到更高的CUDA版本。


不完美解决方案(凑合能用)

一句话版:安装1.12的torch配11.3的CUDA

因为“电脑的CUDA版本向下兼容pytorch所需求的CUDA版本。比如我的CUDA是11.5,那pytorch我勾选11.3就可以。nvidia-smi的cuda版本大于等于你要安装的cuda版本就不会出问题,且你的实际cuda版本以nvcc为准。”(以上引用知乎作者:椿楸 ​的文章CUDA的正确安装/升级/重装/使用方式)

因此,选择去pytorch官网安装支持CUDA11.5版本的torch,如下图所示。详细 安装/重装/卸载 教程可参考上述知乎专栏或者自己另行搜索。安装指令如下,可直接复制使用:

pip install torch==1.12.1+cu113 torchvision==0.13.1+cu113 torchaudio==0.12.1 --extra-index-url https://download.pytorch.org/whl/cu113

用stable_diffusion_webui遇到CUBLAS_STATUS_INVALID_VALUE when calling cublasGemmStridedBatchedExFix 报错文章来源地址https://www.toymoban.com/news/detail-490263.html

不完美解决方案带来的问题!!!

  1. 需要重新安装xformers,因为之前安装的时候torch版本为1.13.1。
  2. stable_diffusion_webui项目可能会有一些奇奇怪怪的bug,因为开发者用的是1.13.1版本开发。(运行的时候会报详细的提示信息,也可通过在参数中加入–skip-version-check进行跳过)
    用stable_diffusion_webui遇到CUBLAS_STATUS_INVALID_VALUE when calling cublasGemmStridedBatchedExFix 报错
都看到这了,点个赞再划走吧

到了这里,关于用stable_diffusion_webui遇到CUBLAS_STATUS_INVALID_VALUE when calling cublasGemmStridedBatchedExFix 报错的文章就介绍完了。如果您还想了解更多内容,请在右上角搜索TOY模板网以前的文章或继续浏览下面的相关文章,希望大家以后多多支持TOY模板网!

本文来自互联网用户投稿,该文观点仅代表作者本人,不代表本站立场。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。如若转载,请注明出处: 如若内容造成侵权/违法违规/事实不符,请点击违法举报进行投诉反馈,一经查实,立即删除!

领支付宝红包 赞助服务器费用

相关文章

觉得文章有用就打赏一下文章作者

支付宝扫一扫打赏

博客赞助

微信扫一扫打赏

请作者喝杯咖啡吧~博客赞助

支付宝扫一扫领取红包,优惠每天领

二维码1

领取红包

二维码2

领红包