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

ImportError: cannot import name ‘InterpolationMode‘ from ‘torchvision.transforms‘踩坑记录
进行查找是有这个模块的。
ImportError: cannot import name ‘InterpolationMode‘ from ‘torchvision.transforms‘踩坑记录
在返回来,查看自己对应版本的官方文档0.8.1
ImportError: cannot import name ‘InterpolationMode‘ from ‘torchvision.transforms‘踩坑记录
是没有的,因此就是版本太低。

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

Original: https://blog.csdn.net/prague6695/article/details/120457004
Author: 在学习的王哈哈
Title: ImportError: cannot import name ‘InterpolationMode‘ from ‘torchvision.transforms‘踩坑记录

原创文章受到原创版权保护。转载请注明出处:https://www.johngo689.com/709099/

转载文章受原作者版权保护。转载请注明原作者出处!

(0)

大家都在看

亲爱的 Coder【最近整理,可免费获取】👉 最新必读书单  | 👏 面试题下载  | 🌎 免费的AI知识星球