RTX40xx 系列显卡提升AI绘图速度方式

matthew8825252 · 发表于 2023-3-29 10:50:26

本帖最后由 matthew8825252 于 2023-3-29 15:22 编辑

https://www.reddit.com/r/StableDiffusion/comments/y71q5k/4090_cudnn_performancespeed_fix_automatic1111/

依据上面网址所述, 先安装Pytorch,将CUDA版本更新到最新版本(目前是CUDA11.8)
如何安装请自行上网搜寻, 这里不多作说明

cudnn 下载网址
https://developer.download.nvidia.com/compute/redist/cudnn/v8.6.0/local_installers/11.8/

下载 cudnn-windows-x86_64-8.6.0.163_cuda11-archive.zip

解压缩后将cudnn-windows-x86_64-8.6.0.163_cuda11-archive\cudnn-windows-x86_64-8.6.0.163_cuda11-archive\bin 目录里面的所有档案复制到 stable diffusion的资料夹\venv\Lib\site-packages\torch\lib, 覆盖掉原本的档案即可

(覆盖之前, 建议先保存一份旧档lib目录)

如果是用大佬的一键安装包, 那先找到site-packages目录, 再往下找site-packages\torch\lib 即可

基本上40xx系列效能愈好的颢卡, 提升速度愈明显, RTX4090 测试后速度提升非常有感,沒詳細作比較,但感覺快了1倍左右,似乎占用的VRAM也会减少