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