美文网首页
M1 配置YOLO环境

M1 配置YOLO环境

作者: 06am | 来源:发表于2022-07-23 22:28 被阅读0次

一、OpenCV

注意,OpenCV使用pip安装即可,conda安装查找不到。
安装这两个

pip install opencv-python
pip install opencv-contrib-python

二、pytorch

运行yolo detect.py报错:
AttributeError: 'Upsample' object has no attribute 'recompute_scale_factor'
原因:pytorch版本过高,一开始安装的1.12.0,支持mps加速。


error

卸载:

conda uninstall pytorch torchvision torchaudio

安装低版本:
低版本官网链接,conda查找不到,还是使用pip安装。

pip install torch==1.10.1 torchvision==0.11.2 torchaudio==0.10.1

相关文章

网友评论

      本文标题:M1 配置YOLO环境

      本文链接:https://www.haomeiwen.com/subject/mtxiirtx.html