CUDA 12.1 + cuDNN 8.8
到 NVIDIA 官网下载这两个包。
把 cuDNN 中的 bin\*.dll,覆盖到 venv\Lib\site-packages\torch\lib。 cuda
包里面解压出来 cublas64_12.dll、cublasLt64_12.dll,也放到这个目录。
---------------------------
以上内容引用:优化 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