ModuleNotFoundError: No module named 'torchvision.models.mobilenetv2' 에러해결

개발환경

 

- ubuntu18.04

- python3.7

 

 

1. Problem

 

model = torch.hub.load('pytorch/vision:v0.10.0', 'wide_resnet50_2', pretrained=True)

wide_resnet을 학습시키려고 할 때

ModuleNotFoundError: No module named 'torchvision.models.mobilenetv2'

모듈을 찾을 수 없다는 error가 발생

 

 

 

2. Solution

 

정확한 이유는 알 수 없으나

stackoverflow의 해결방법은

pytorch/vision:v0.10.0 -> pytorch/vision:v0.8.0