python环境安装 PyTorch 2.0 和 xformers 0.0.17

AI   2024-01-12 09:49   565   0  

下面操作以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

Install PyTorch 2.0

1
pip install -U torch torchvision torchaudio numpy  --index-url https://download.pytorch.org/whl/cu118

Install xformers 0.017

1
pip install xformers==0.0.17rc482 

Install Accelerate

1
pip install -U accelerate==0.15.0

Replace CUDNN's DLL Files

下载 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


博客评论
还没有人评论,赶紧抢个沙发~
发表评论
说明:请文明发言,共建和谐网络,您的个人信息不会被公开显示。
QQ
微信
打赏
扫一扫