文章目录
- 1.遇到问题
- 2.解决方式
1.遇到问题
在调试yolov5时，我们发现在运行时发现提示是：using CPU
2.解决方式
- 将device中的default =“CPU”改成默认“ ”；注意：在模块：if name == ‘main’:中
更改前：
parser.add_argument('--device', default='CPU', help='cuda device, . 0 or 0,1,2,3 or cpu')
更改后：
parser.add_argument('--device', default='', help='cuda device, . 0 or 0,1,2,3 or cpu')