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/ 已有版本页面找到相应版本的安装命令,

Python – 运行YOLOv5出现AttributeError: module ‘torchvision’ has no attribute ‘ops’-StubbornHuang Blog

比如这里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'错误了。