![](https://img-blog.csdnimg.cn/20201014180756724.png?x-oss-process=image/resize,m_fixed,h_64,w_64)
YOLOv5
文章平均质量分 71
wind.liao
这个作者很懒,什么都没留下…
展开
-
Windows10使用cuda加速YOLOv5训练
查看自己主机的显卡信息,比如我的显卡是:NVIDIA GeForce GTX 1060 6GB需要安装的工具列表:显卡驱动 (v465.89): cuda v11.3.1 cuDNN v8.2.1 pytorch v1.10.1步骤一:安装显卡驱动使用下面的链接查找cuda版本对应的显卡驱动(此步骤千万不要弄错了)https://docs.nvidia.com/cuda/cuda-toolkit-release-notes/NVIDIA 驱动程序下载 - 高级搜索原创 2021-12-31 09:04:26 · 3605 阅读 · 1 评论 -
Windows 10安装YOLOv5训练及验证狗狗照片
1. Install YOLOv5$git clone https://github.com/ultralytics/yolov5$cd yolov5$pip install -r requirements.txt2. 训练数据集(已标签过)链接:https://pan.baidu.com/s/1kMckA7OAi-UZdxgdD9ys7w提取码:adnb3. 执行train.py开始训练狗狗照片$python train.py --data data/dogs.yaml.原创 2021-12-31 08:43:17 · 1550 阅读 · 0 评论