使用CUDA过程中出现异常

本文主要是介绍使用CUDA过程中出现异常，希望对大家解决编程问题提供一定的参考价值，需要的开发者们随着小编来一起学习吧！

使用：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

这篇关于使用CUDA过程中出现异常的文章就介绍到这儿，希望我们推荐的文章对编程师们有所帮助！

---