Autodistill Yolov5 issue on colab tutorial

autodistill / autodistill-yolov5

YOLOv5 module for use with Autodistill.

https://docs.autodistill.com

GNU Affero General Public License v3.0

1 stars 2 forks source link

Open cadrev opened 1 year ago

cadrev commented 1 year ago

Hello, I've been meaning to run YoloV5 for autodistill and I'm using the YoloV8 google colab reference.

Whenever I use YoloV5 i keep getting this error

AssertionError: Invalid CUDA '--device cuda' requested, use '--device cpu' or pass valid CUDA device(s)

capjamesg commented 1 year ago

@cadrev Can you share a link to the Colab you are using?