出现这个报错之后我就去百度搜索,然后发现很多报错都是因为pytorch的问题,在看了很多人给的解决办法之后我认为都是因为版本问题所导致的,或者在安装的时候出现了问题。
首先我是运行了一份yolo代码它提示我是没有torchvision,没有就去安装嘛,没有什么大惊小怪的。然后我去pip安装了一下
因为个人版本是cuda10.0,和pytorch 1.2.0,所以我就按照官网这么去安装了
(链接https://pytorch.org/get-started/previous-versions/)
然后安装的时候提示我的是,找不到0.4.0的版本torchvision,其实这个时候我就很懵逼了,为啥会找不到,但是它提供了一个选项是有0.4.1,我寻思着这俩应该差不多,就按照了0.4.1,结果就报错了如题的错误。
对于pytorch 1.2.0版本相对于的是torchvision 0.4.0,pip在线安装不了我就去离线安装。
我观察到下面俩个命令提供了一个网址,我就去查看了一这个网址
https://download.pytorch.org/whl/torch_stable.html
果不其然找到了一个令我满意的东西,于是直接下载whl文件,进行本地pip安装。
pip本地安装方法参考我这篇博客python3 PIL安装教程 PIL替代pillow安装教程
安装完成之后运行成功