- 博客(5)
- 收藏
- 关注
原创 YOLOv8无法使用GPU训练的问题
经过排查,YOLOv8在安装依赖项的时候直接pip install ultralytics,全部集成在了ultralytics里面,但是torch的版本可能是最新的,而你的cuda如果在11.2以下就无法正常调用cuda了。在你的train.py代码中加入:print(torch.cuda.is.available())后运行,如果得到的是false,证明你没有成功的调用cuda来加速你的模型训练。2.下载指定版本的1.10的torch(我这里还使用了清华镜像源进行加速)然后输入y表示yes。
2024-05-10 15:31:21 1721
空空如也
空空如也
TA创建的收藏夹 TA关注的收藏夹
TA关注的人