I have torchvision 0.5.0 and torch 1.4.0. importing fastai gives the following error

I have torchvision 0.5.0 and torch 1.4.0. importing fastai gives the following error

from fastai.vision import *
from fastai.metrics import error_rate

Running this from the notebook of first lesson gives the following error. Any ideas?

ImportError Traceback (most recent call last)
in
----> 1 from fastai.vision import *
2 from fastai.metrics import error_rate

~/anaconda3/envs/pytorch/lib/python3.7/site-packages/fastai/vision/init.py in
1 from … import basics
2 from …basics import *
----> 3 from .learner import *
4 from .image import *
5 from .data import *

~/anaconda3/envs/pytorch/lib/python3.7/site-packages/fastai/vision/learner.py in
4 from …basic_data import *
5 from .image import *
----> 6 from . import models
7 from …callback import *
8 from …layers import *

~/anaconda3/envs/pytorch/lib/python3.7/site-packages/fastai/vision/models/init.py in
1 from .xresnet import *
----> 2 from torchvision.models import ResNet,resnet18,resnet34,resnet50,resnet101,resnet152
3 from torchvision.models import SqueezeNet,squeezenet1_0,squeezenet1_1
4 from torchvision.models import densenet121,densenet169,densenet201,densenet161
5 from torchvision.models import vgg11_bn,vgg13_bn,vgg16_bn,vgg19_bn,alexnet

ImportError: cannot import name ‘ResNet’ from ‘torchvision.models’ (unknown location)

However this code runs fine

form torchvision.models import *

1 Like