AI画图01

邱秋 • 2023年12月07日 • 阅读:70 • AI

安装 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% 的错误

我,秦始皇,打钱!