Python – 运行YOLOv5出现AttributeError: module ‘torchvision’ has no attribute ‘ops’
1 运行YOLOv5出现AttributeError: module 'torchvision' has no attribute 'ops'
今天在下载完Yolov5的依赖包之后，使用detect.py运行出现了
AttributeError: module ‘torch‘ has no attribute ‘ops‘
这个问题出现的原因是torchvision与torch的版本不匹配造成的。
之前安装pytorch命令是
conda install pytorch=1.8.0 torchvision torchaudio cudatoolkit=10.2 -c pytorch
没有指定torchvision的版本。
这里推荐一下正确安装pytorch的方法，应该从官方网站的https://pytorch.org/get-started/previous-versions/ 已有版本页面找到相应版本的安装命令，
比如这里Pytorch1.8.1+CUDA10.2的安装命令为
conda install pytorch==1.8.1 torchvision==0.9.1 torchaudio==0.8.1 cudatoolkit=10.2 -c pytorch
使用这个命令安装之后，就没有出现AttributeError: module 'torchvision' has no attribute 'ops'错误了。
本文作者：StubbornHuang
版权声明：本文为站长原创文章，如果转载请注明原文链接！
原文标题：Python – 运行YOLOv5出现AttributeError: module ‘torchvision’ has no attribute ‘ops’
原文链接：https://www.stubbornhuang.com/2086/
发布于：2022年04月11日 10:55:08
修改于：2023年06月26日 20:18:29
声明：本站所有文章，如无特殊说明或标注，均为本站原创发布。任何个人或组织，在未征得本站同意时，禁止复制、盗用、采集、发布本站内容到任何网站、书籍等各类媒体平台。如若本站内容侵犯了原著者的合法权益，可联系我们进行处理。
文章末尾
评论
50