使用:yolo detect train data=SKU-110K.yaml model=yolov8n.pt epochs=100 imgsz=640 device=0,1
出现错误
UserWarning: CUDA initialization: CUDA unknown error - this may be due to an incorrectly set up e
nvironment, e.g. changing env variable CUDA_VISIBLE_DEVICES after program start. Setting the available devices to be zero.
解决办法:
sudo apt-get install nvidia-modprobe