QQ扫一扫联系
下面操作以Windows 环境为主,首先以管理员身份打开windows powershell,更改执行策略权限调整为: remotesigned
输入指令:
1 | Set-ExecutionPolicy -ExecutionPolicy RemoteSigned -Scope LocalMachine |
或者:
1 | set -ExecutionPolicy RemoteSigned |
查看变更:
1 | get-ExecutionPolicy |
进入venv\Scripts 目录,运行.\activate
PyTorch 2.0 需要CUDA 11.8,先去下载CUDA Toolkit 11.8。
1 | pip install - U torch torchvision torchaudio numpy - - index - url https: / / download.pytorch.org / whl / cu118 |
1 | pip install xformers = = 0.0 . 17rc482 |
1 | pip install - U accelerate = = 0.15 . 0 |
下载 cudnn 8.7.0把bin 的DLL 文件更换到venv\Lib\site-packages\torch\lib,重新启动stable diffusion 完成。
Optimization
使用Torch 2.0 可无须配搭xformers,加上优化参数:
1 | --opt-sdp-attention |
或
1 | --opt-sdp-no-mem-attention |