在运行resNeSt代码的时候,有一个报错。

ImportError: cannot import name ‘InterpolationMode’ from ‘torchvision.transforms’ (C:\ProgramData\Anaconda3\lib\site-packages\torchvision\transforms_init_.py)

但是网上都找不到相关解决办法。
于是就自己排除,记录以下。
出错代码如下:

from torchvision.transforms import Resize,InterpolationMode

先确定自己的torchvision,toch的版本。
使用

import torchvision
print("Torchvision Version: ",torchvision.__version__)

import torch
print(torch.__version__)

输出为:

Torchvision Version:  0.8.1
1.7.0

然后找到最新的官方文档,查看相应的模块
https://pytorch.org/vision/0.9/transforms.html
在这里插入图片描述
进行查找是有这个模块的。

在返回来,查看自己对应版本的官方文档0.8.1
在这里插入图片描述
是没有的,因此就是版本太低。

然后新建conda环境重新安装torch和torchvision即可。
PYTORCH与TORCHVISION的更新,卸载和秒安装(看这一篇就够了)
参考这个

Logo

华为开发者空间,是为全球开发者打造的专属开发空间,汇聚了华为优质开发资源及工具,致力于让每一位开发者拥有一台云主机,基于华为根生态开发、创新。

更多推荐