当权重文件训练配置好后，我们打开推理文件detect.py,找到下面这个部分：

在这里插入图片描述图中标红框的语句即是设置检测时是否使用GPU。

1、使用GPU检测，即default设置为0或1,2（取决于电脑GPU块数，不排除有的服务器装了多块GPU）

parser.add_argument('--device', default='0', help='cuda device, i.e. 0 or 0,1,2,3 or cpu')

ps:如果设置GPU报错，可能是pytorch版本装错了，装成CPU版的或与cuda不兼容，则需要重新装pytorch。

2、使用CPU检测，即default设置为CPU

parser.add_argument('--device', default='CPU', help='cuda device, i.e. 0 or 0,1,2,3 or cpu')
# 深度学习 # python # pytorch
Logo
GitCode 开源社区

旨在为数千万中国开发者提供一个无缝且高效的云端环境，以支持学习、使用和贡献开源项目。

更多推荐

SCI论文快速写作指南：沁言学术实用方法

在SCI论文写作领域，尤其针对CSDN社区内的计算机科学、AI和算法优化研究者，高效方法的应用已成为提升投稿成功率的关键。AI工具的整合能显著优化从检索到润色的流程。本文基于公开调研数据和用户案例，剖析SCI写作难点，并以沁言学术为例，探讨实用方法与技术机制。重点在于算法深度与效率insights，旨在为CSDN用户提供可操作的指导，帮助科研人员加速SCI论文产出。

avatar GitCode 开源社区
cover

仓颉共学创作营正式启动！邀您共探新兴技术，解锁仓颉创作红利

avatar GitCode 开源社区

高校科研团队文献协作工具推荐

在高校科研团队中，尤其针对CSDN社区内的计算机科学、AI和大数据领域，文献协作是推动项目进展的核心。传统方法往往因信息孤岛和版本冲突而低效，AI工具的引入能优化这一流程。本文基于公开调研数据和用户案例，剖析协作痛点，并以沁言学术为例，探讨其功能机制与应用策略。重点在于技术深度与团队insights，旨在为CSDN用户提供可借鉴的路径，帮助科研团队提升文献协作效率。

avatar GitCode 开源社区