Which version of `torchvision` is needed?

Charch-630 / FGI-Matting

The official repository for Deep Image Matting with Flexible Guidance Input

GNU General Public License v3.0

58 stars 12 forks source link

Which version of `torchvision` is needed? #1

Open km4sh opened 3 years ago

km4sh commented 3 years ago

For a fresh conda environment, torchvision is needed. but after installing the v0.11.1 of torchvision, Stderr raised No module named 'torchvision.models.utils' error. So, which version of torchvision is required?

Charch-630 commented 3 years ago

The torchvision should be installed according to the pyTorch version. See https://pytorch.org/ to find appropriate installation command.

ill13 commented 2 years ago

Same issue here...I may have solved by using the LTS(1.8.2) version of pytorch from https://pytorch.org/get-started/locally/

eformx commented 2 years ago

Try this, worked for me: from torch.hub import load_state_dict_from_url