首页yolov5训练如何指定gpu

yolov5训练如何指定gpu

时间: 2024-05-22 17:08:49 浏览: 341

yolov5s nnie.zip

yolov5s nnieyolov5-nnieyolov5s nnieYOLOv5 pytorch -> onnx -> caffe -> .wk 1、模型是yolov5s,将focus层替换成stride为2的卷积层。reshape和permute层也做了调整。具体的修改过程可以参考这个大佬的文章https://blog.csdn.net/tangshopping/article/details/1100386052、模型是在hi3559av100上跑的，mapper版本是1.2。3、用法mkdir buildcd buildcmake -DCMAKE_TOOLCHAIN_FILE=../hi3559.toolchain.cmake ..make -j4./yolo_nnie参考https://blog.csdn.net/tangshopping/article/details/110038605watermelooon/nnie_yolohttps://github.com/ultralytics/yolov5https://githu

在使用YOLOv5训练时，可以通过指定`--device`参数来选择使用的GPU设备，例如： ``` python train.py --epochs 30 --data coco.yaml --weights yolov5s.pt --device 0 ``` 其中，`--device 0`表示使用第一个GPU设备。如果你有多个GPU设备，可以使用不同的数字来指定。此外，如果不想使用GPU进行训练，可以将该参数设置为-1。

阅读全文