主要原因是torchvision的版本与pytorch版本不兼容,选择和pytorch版本兼容的torchvision版本即可。

一、查看pytorch的版本:在python IDE中输入,

import torch
print(torch.__version__)

可以看到我的pytorch版本为1.7.1:

二、选择相匹配的torchvision版本:

可以到pytorch官网上根据自己安装的pytorch确认应该安装的torchvision版本,比如pytorch1.7.1就应该对应torchvision0.8.2:

这样就可以成功安装啦!

 

 

Logo

更多推荐