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的更新，解除安裝和秒安裝（看這一篇就夠了）

參考這個