1.第一步:升级 PyTorch 到 2.2.0 及以上(推荐 2.3.0 或 2.4.0)
pip install --upgrade "torch>=2.2.0" --index-url https://download.pytorch.org/whl/cu121
2.第二步:升级 torchvision、torchaudio、xformers 到与新 torch 兼容的版本
pip install --upgrade torchvision torchaudio xformers --index-url https://download.pytorch.org/whl/cu121
3.确认依赖版本一致
import torch
import torchvision
import torchaudio
import xformers
print(torch.__version__)
print(torchvision.__version__)
print(torchaudio.__version__)
print(xformers.__version__)