ImportError: cannot import name ‘InterpolationMode‘ from ‘torchvision.transforms‘踩坑记录

在运行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的更新，卸载和秒安装（看这一篇就够了）

参考这个