我今天遇到了这个问题。
网上大部分的博客,都要直接让torch和torchvision降级去下载,我来说点人话。抄袭那些真的有意思吗?呵呵。

我的问题具体触发的原因呢,是我本来用的是cpu的torch,前些阵子直接换成了gpu的torch,由于是直接upgrade的,尽管torch本身是更新成了torch1.7.0+cu110,但是torchvision仍然是cpu版本的。
于是在运行之前写好的代码的时候,就出现了上面的错误。

所以,说点人话,那就是你的torch和torchvision的版本要对应。我并没有用大部分博客的方法,我是直接升级了torchvision的版本,升级到对应于我torch1.7.0+cu110的版本。
升级的注意点在于,你不能直接
pip insatll upgrade xxx,因为没有资源。
我是从pytorch官网上下载的,也就是
在这里插入图片描述使用的这个,升级完成后,完美运行。

Logo

昇腾计算产业是基于昇腾系列(HUAWEI Ascend)处理器和基础软件构建的全栈 AI计算基础设施、行业应用及服务,https://devpress.csdn.net/organization/setting/general/146749包括昇腾系列处理器、系列硬件、CANN、AI计算框架、应用使能、开发工具链、管理运维工具、行业应用及服务等全产业链

更多推荐