完美解决AttributeError: module ‘torchvision.models‘ has no attribute ‘detection‘

报错信息

使用torchvision报错如下：

错误原因

报错翻译过来是：

torchvision.models' 没有属性 'detection'

错误产生的原因是找不到该属性，但是别人都能这么用，说明可能是我们的版本太低了。

解决方案

如果你直接使用python环境，直接使用下面的命令就可以解决：

pip install torchvision==0.4.1

如果使用anaconda，并且电脑中只有一个python环境，也可以直接使用上面的命令解决。

如果你用anaconda创建了多个环境，就需要在对应的环境做设置了。

进入anaconda我们自己创建的环境，找到我们要使用的环境：我们发现，它的版本是0.2.1，而且没有办法升级。

也就是直接使用anaconda是不能升级的。

我们打开anaconda的Prompt：

使用如下命令：

conda info -e

conda activate [环境]  //例如：conda activate pytorch

查看一下我们电脑的环境，并进入到我们所用的环境：

查看环境

进入环境

然后使用命令安装：

pip install torchvision==0.4

如果你不知道要安装哪个版本，可以随便乱输入一个不存在的版本，他会给你提示：