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__)
Logo

火山引擎开发者社区是火山引擎打造的AI技术生态平台,聚焦Agent与大模型开发,提供豆包系列模型(图像/视频/视觉)、智能分析与会话工具,并配套评测集、动手实验室及行业案例库。社区通过技术沙龙、挑战赛等活动促进开发者成长,新用户可领50万Tokens权益,助力构建智能应用。

更多推荐