图形深度学习之-数据炼丹(训练)

上篇文章我们介绍了如何对需要训练的数据进行标注，详细可看：

图形深度学习之炼丹必备技能-数据标注https://blog.csdn.net/Tangrongji/article/details/135607678?spm=1001.2014.3001.5501有了数据标注的基础，接下来就可以对标注的数据进行模型训练了，我这里使用的训练模型和环境做个简单介绍：

• Intel(R) Core(TM) i7-7700K CPU @ 4.20GHz 4.20 GHz

• GPU-1060-6GB

• 48G内存

• Python 3.10.x

• CUDA - 11.8

• 训练模型：YoloV8

• TensorRT - 8.6.1.6

• cudNN - 8.9.6.50

这里如果不知道如何安装 CUDA、TensorRT和cudNN的，可以参考之前的文章：