一 安装 Python 或者 Anaconda,我安装了Anaconda就以 conda 为例。
安装完conda,创建一个新环境,我使用了python 3.10
conda create -n py310 python=3.10 anaconda
切换环境
conda deactivate
conda activate py310
安装pip
conda install pip
二 在 conda 中安装需要的库
安装 pytorch:
conda install pytorch torchvision torchaudio pytorch-cuda=11.8 -c pytorch -c nvidia
这里我已经安装了cuda toolkit 11.8。
安装 transformers:
conda install transformers
这个安装可能会缺少模块,可以使用下面的安装:
pip install git+https://github.com/huggingface/transformers.git
安装 diffusers:
conda install -c conda-forge diffusers
其他会需要的库:
pip install timm
pip install accelerate
三 几个很好玩的例子:
1 中