原因
在跑Tracking-Anything-with-DEVA模型的时候pytorch环境没有配置好,显卡驱动cuda version11.7装成了11.8,pytorch也需要从2.1.0换到2.0.1,导致torchvision不匹配。
根据报错信息前往github.com/pytorch/vision
查看已安装版本
import torch
torch.__version__ # 输出 2.0.1
import torchvision
torchvision.__version__ # 输出 0.16.0
版本不对,重新安装
1.卸载原有版本
conda卸载比较慢所以用了pip。
pip uninstall torchvision==0.16.0
一定要先卸载再安装,第一次尝试直接安装以为会覆盖更新结果并没有,后来执行了两次才卸干净(抹泪)
2.安装torchvision
先找到合适的版本:https://download.pytorch.org/whl/torchvision/
我这里是linux+py38+cu11.7
pip install https://download.pytorch.org/whl/cu117/torchvision-0.15.2%2Bcu117-cp38-cp38-linux_x86_64.whl
3.验证
import torchvision
torchvision.__version__
输出‘ 0.15.2+cu117’就成功力
(参考) torch-torchvision-python对应版本