一、环境搭建
Windows, Python-3.9.7 torch-1.8.0+cu111 CUDA:0 (NVIDIA GeForce 940M, 1024MiB),YOLOv5
环境配置尝试了两种方式:
conda 下配置,较容易实现;
python下pip 方式,从官网参考的命令没有与CUDA11.2匹配的选项,只能选择手动下载torch-1.9.1+cu111-cp39-cp39-win_amd64.whl,通过pip 命令安装过程中又自动更改为torch-1.8.0.;然后手动安装torchvision和torchaudio。
参考了这篇文章:【环境配置】RTX 3090+cuda11.2+python39+pytorch1.8.0_燕策西的博客-CSDN博客_3090显卡cuda版本
二、实验步骤
主要参考了:
YOLOv5解析 | 第二篇:用YOLOv5训练自己的数据集(超级详细完整版)_K同学啊的博客-CSDN博客_yolov5 训练自己的数据集
踩