训练yolov5中遇到的问题

环境问题是大家遇到最多的问题,用cmd运行pip list查看所用环境是否正确
在这里建议大家使用:

python3.7
torch
1.6.0+gpu(如果没有gpu就直接torch1.6.0)
torchvision
0.7.0+gpu(同上)
cuda10.2
问题出的最多的就是numpy,这个不要单独装,他是和torch相关联的,如果版本对应不上可能会报错,我这里使用的是numpy
1.19.5
**一旦发现报错首先看环境对不对!!!**环境不对改代码也是白搭

这是我的环境
在这里插入图片描述

猜你喜欢

转载自blog.csdn.net/weixin_43969718/article/details/113108224