1.环境配置
-
创建虚环境
conda create -n sd3.5 python=3.10
-
Pytorch(>2.0)
conda install pytorch==2.2.2 torchvision==0.17.2 torchaudio==2.2.2 pytorch-cuda=12.1 -c pytorch -c nvidia
-
Jupyter能使用Anaconda虚环境
conda install ipykernel python -m ipykernel install --user --name sd3.5 --display-name "SD3.5"
-
安装transformer和tokenizer
pip install transformers==4.38.2 pip install tokenizers==0.15