根据深入浅出Yolov5之自有数据集训练超详细教程进行学习过程中,
强烈建议用conda创建新的虚拟环境,重新在有requirements.txt的目录下pip install -r requirements.txt,可以避免很多问题
安装pytorch可能出现的问题。
出现问题1: 在用pip install -r requirements.txt
的时候,安装的是最新版本的torch、和torchvision,运行detect.py
时报错 ImportError: Failed to load PyTorch C extensions:,检查到可能情况如下:
- 本来的python库中已安装
numpy
,在用requirements.txt安装时又覆盖安装了新的numpy
,导致存在多个numpy。
- 解决办法:
pip uninstall numpy
,出现询问是否删除,选择 是,此时卸载后只剩一个numpy
- 默认安装的最新版torch,但是用不了。
- 解决办法:卸载torch,torchvision,torchaudio。重新安装旧版本,经试验,
torch== 1.9.0 + torchvision ==0.10.0 + torchaudio == 0.9.0
是可以的,对应关系见官方repo - 卸载:
pip uninstall torch
,pip uninstall torchvision
,pip uninstall torchaudio
,来源博客
出现问题2: ModuleNotFoundError: No module named ‘kiwisolver’
解决办法:卸载kiwi solver,pip uninstall kiwisolver
,再重新安装,pip install kiwisolver
.
学习过程中注意:
main.py
中,将阶段一的注释可以不用打开,只要跑阶段二阶段三就可以了。不然阶段一会弹出很多张图片而且linux命令行中退不掉!