赞
踩
一、更新显卡驱动
二、Linux安装CUDA和cudnn:
一定要版本匹配,也要和驱动的版本匹配
在这里我用的是CUDA10.0和cudnn7.6.5
【CUDA】Ubuntu系统如何安装CUDA保姆级教程(2022年最新)_ubuntu安装cuda_自牧君的博客-CSDN博客
三、配置conda并安装yolov8
最新YOLOv8(2023年8月版本)安装配置!一条龙傻瓜式安装,遇到问题评论区提问_yolo安装-CSDN博客
PS:经尝试
pip config set install.trusted-host mirrors.aliyun.com
的镜像源不是太好,如果有同样的问题可以换一个镜像源
写完代码后在终端输入
- conda activate yolov8
- cd ultralytics/ultralytics-main
- python yolov8_train.py #训练的.py文件名
- python yolov8_dect.py #检测的.py文件名
如果跑不动,中间显示“已杀死”或训练非常缓慢
可能是
①没有用上GPU,输入:
nvidia-smi
显示
可以查看python是否占用GPU中,以及GPU使用了多少
没用上的原因可能是:显卡驱动版本不匹配
②交换空间满了
打开系统监视器
如果满了可以增加交换空间,详见Ubuntu上增加交换空间-CSDN博客
Copyright © 2003-2013 www.wpsshop.cn 版权所有,并保留所有权利。