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

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

报错信息

使用torchvision报错如下:

20201130105800397.png

错误原因

报错翻译过来是:

torchvision.models’ 没有属性 ‘detection’

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

解决方案

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

  1. 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:

使用如下命令:

  1. conda info -e
  2. conda activate [环境] //例如:conda activate pytorch

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

20201130110829927.png 查看环境

20201130111134912.png 进入环境

然后使用命令安装:

  1. pip install torchvision==0.4

watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L3NodWl5aXhpbg_size_36_color_FF47FF_t_70

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

2020113011145689.png

发表评论

表情:
评论列表 (有 0 条评论,291人围观)

还没有评论,来说两句吧...

相关阅读