M1 配置YOLO环境

一、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
©著作权归作者所有,转载或内容合作请联系作者
平台声明:文章内容(如有图片或视频亦包括在内)由作者上传并发布,文章内容仅代表作者本人观点,简书系信息发布平台,仅提供信息存储服务。

推荐阅读更多精彩内容