优化 Stable Diffusion --opt-sdp-attention

[复制链接]
查看486 | 回复0 | 2023-8-16 15:13:17 | 显示全部楼层 |阅读模式

CUDA 12.1 + cuDNN 8.8


  1. 到 NVIDIA 官网下载这两个包。
  2. 把 cuDNN 中的 bin\*.dll,覆盖到 venv\Lib\site-packages\torch\lib。 cuda
  3. 包里面解压出来 cublas64_12.dll、cublasLt64_12.dll,也放到这个目录。
  4. ---------------------------
复制代码
以上内容引用:优化 Stable Diffusion,提高出图速度 - 鱼头的文章 - 知乎
https://zhuanlan.zhihu.com/p/621457549
以下自己的:
4090显卡可以使用这个:
set PYTHON=%CD%\py\python.exe
set GIT=%CD%\git\bin\git.exe
set GIT_PYTHON_GIT_EXECUTABLE=%CD%\git\bin\git.exe
set GIT_PYTHON_REFRESH=quiet
set TRANSFORMERS_CACHE=%CD%.cache\huggingface\transformers
set HUGGINGFACE_HUB_CACHE=%CD%.cache\huggingface\hub
set COMMANDLINE_ARGS= --autolaunch --opt-sdp-attention --deepdanbooru
%python% launch.py
pause

来源:https://blog.csdn.net/qq_42944545/article/details/130440017
免责声明:如果侵犯了您的权益,请联系站长,我们会及时删除侵权内容,谢谢合作!
回复

使用道具 举报

您需要登录后才可以回帖 登录 | 立即注册

本版积分规则