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

小鱼儿 2022-12-25 02:54 242阅读 0赞

# 报错信息 #

使用torchvision报错如下：

![20201130105800397.png][]

# 错误原因 #

报错翻译过来是：

> torchvision.models' 没有属性 'detection'

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

# 解决方案 #

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

pip install torchvision==0.4.1

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

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

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

![watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L3NodWl5aXhpbg_size_66_color_FF47FF_t_70][]

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

我们打开anaconda的Prompt：

使用如下命令：

conda info -e
    
    conda activate [环境]  //例如：conda activate pytorch

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

![20201130110829927.png][] 查看环境

![20201130111134912.png][] 进入环境

然后使用命令安装：

pip install torchvision==0.4

![watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L3NodWl5aXhpbg_size_36_color_FF47FF_t_70][]

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

![2020113011145689.png][]

[20201130105800397.png]: /images/20221120/220a62c96e8e49e69ff8556c8f8ec2f5.png
[watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L3NodWl5aXhpbg_size_66_color_FF47FF_t_70]: /images/20221120/9d83cf99cd164f72bc17105c3dfb8fa7.png
[20201130110829927.png]: /images/20221120/0fa7f8f913004b9e87d0e3c8f71ec5b2.png
[20201130111134912.png]: /images/20221120/e2c17c92f0c543d5a5c8fd1cbd5767c8.png
[watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L3NodWl5aXhpbg_size_36_color_FF47FF_t_70]: /images/20221120/1eb9db015c4048918698ff1d32341f3e.png
[2020113011145689.png]: /images/20221120/a6bf01a135cd40d1bff830b2338800d8.png