在上文中，笔者介绍了使用google公司提供的免费GPU资源colab来对大量的自定义数据集进行模型训练。该方法虽然简单好用，但是存在以下几方面的短板问题：
一是需要通过虚拟服务器做为跳板机来访问，总体操作起来非常繁杂。
二是需要将大量的数据上传缓慢，管理和使用非常不友好，特别是我们分批上传文件时，需要后期逐个整理文件，非常不友好。
三是免费使用gpu的额度有限制，也就是说当你频繁使用colab做训练的时候，会提示配额已满，甚至我们耗费很长时间进行训练的时候，突然运行报错，实际上就是配额慢了，造成前期工作白费了。
为此，在我们实际的项目需求中，仍然需要掌握如何在windows环境下，调用本地电脑的GPU资源进行模型的训练。为了确保您的电脑中显卡资源可以被成功调用，您需要首先完成电脑显卡驱动和CUDA、torch等的安装准备工作。完成之后，便可调用本地电脑的GPU资源了，具体的代码与基于linux环境的colab上运行的代码类似，但有些许不同，笔者将逐一讲解。
1 、首先查看自己的显卡型号
根据nvcc-smi查到自己的显卡型号，如下图所示。
本人的电脑显卡型号为：GeForce GT 730文章来源：https://www.toymoban.com/news/detail-726427.html
2、查看显卡算力
可以通过以下链接查找
http://www.5ityx.com/cate100/155907.html
可以看到我的显卡算力是3.5文章来源地址https://www.toymoban.com/news/detail-726427.html
到了这里，关于04训练——基于YOLO V8的自定义数据集训练——在windows环境下使用pycharm做训练-1总体步骤的文章就介绍完了。如果您还想了解更多内容，请在右上角搜索TOY模板网以前的文章或继续浏览下面的相关文章，希望大家以后多多支持TOY模板网！
