0
点赞
收藏
分享

微信扫一扫

torchvision与torch的版本冲突

年迈的代码机器 2022-04-05 阅读 164

在import torchvision时报错:UserWarning: Failed to load image Python extension: [WinError 127]


解决

在晚上查询之后,问题在于自己的torchvision版本与torch版本不对应。

首先查询自己的torch和torchvision版本

import torch
print(torch.__version__)

import torchvision
print(torchvision.__version__)

因为我自己的版本是比较新的,需要到Github上面查询最近的两者版本之间的关系:

链接: https://github.com/pytorch/vision#installation.

版本对应关系
我的版本关系为:
torch1.11.0

对应的torchvision版本为:
torchvision0.12.0

由于镜像下载老是出一堆问题,直接下载过于缓慢,我选择通过whl文件安装,网址为:

https://download.pytorch.org/whl/torch_stable.html

通过pip卸载重装之后,再次运行:

重装之后运行

至此问题解决

举报

相关推荐

0 条评论