AI画图01

安装 cuda

下载链接：https://developer.nvidia.com/cuda-toolkit（注意GPU与cuda的版本对应）

python

https://www.python.org/ 安装最新版本

conda

可选

安装 pytorch

从这里找到对应的版本 ：https://pytorch.org/get-started/locally/ 下面两个是最新的版本2选1

# 121版本
pip install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu121
# 118版本
pip install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu118

安装 xFormers

这里 ：https://github.com/facebookresearch/xformers 2选1

# cuda 11.8 version
pip3 install -U xformers --index-url https://download.pytorch.org/whl/cu118
# cuda 12.1 version
pip3 install -U xformers --index-url https://download.pytorch.org/whl/cu121

版本对应上了，后面会减少80% 的错误