【亲测】yolo v8训练自己的数据集(超详细)_yolov8 gpu训练

作者：Gausst松鼠会 | 2024-02-28 22:32:00

踩

yolov8 gpu训练

利用YOLO V8训练自己数据集之前确保自己环境是否安装正确，本人是用1050 Ti显卡（GPU）＋miniconda下创建的YOLO V8环境来训练的。

安装CUDA+显卡驱动请参照：

【亲测】ubuntu20.4 显卡驱动+CUDA11.8一起安装_Coding_C++的博客-CSDN博客

利用miniconda安装yoloV8的环境请参照：

conda activate yoloV8

在激活的环境下输入python 调起python的环境，然后输入如下命令

import ultralytics

ultralytics.checks()

此时注意以下信息：

Ultralutics YOLOv8.0.181 是YOLO的版本号

python 版本是3.11.4

torch版本是2.0.1

显卡是 NVIDIA GeForce GTX 1050 Ti, 4037MiB（显存大小）

yolo settings

声明：本文内容由网友自发贡献，不代表【wpsshop博客】立场，版权归原作者所有，本站不承担相应法律责任。如您发现有侵权的内容，请联系我们。转载请注明出处：https://www.wpsshop.cn/w/Gausst松鼠会/article/detail/162260