device handling in yolov8.py: device=device? about autodistill-yolov8

[adriansletten]

I have the same issue when running this notebook in google colab. No matter what I change the device to I always get the following error:
ValueError: Invalid CUDA 'device=cuda' requested. Use 'device=cpu' or pass valid CUDA device(s) if available, i.e. 'device=0' or 'device=0,1,2,3' for Multi-GPU.

[adriansletten]

A quick fix is to import version 0.1.0 from pip.
pip install autodistill-yolov8==0.1.0

[capjamesg]

Apologies for the friction you ran into getting started. I pushed a change to enable mps acceleration support for YOLOv8 training in Autodistill but the default device configuration wasn't set up properly.

Upgrade to autodistill-yolov8 0.1.3:

pip install --upgrade autodistill-yolov8==0.1.3

Use this code to train:

target_model.train(DATA_YAML_PATH, epochs=50, devices=[0,1...])

Where devices is the list of CUDA device IDs you want to use in training.