【手把手教你从零开始YOLOv8-入门篇】YOLOv8 环境安装

1,221 阅读2分钟

环境安装

一、前置环境安装-miniconda

1. miniconda 安装

2. conda环境创建

  • 命令:
conda create -n yolov8 python=3.8
  • 明确指定版本,否则可能会因版本过高导致有包装不上

3. Pypi配置国内源

二、前置环境安装-pytorch

1. pytorch安装

  • 官方地址: pytorch.org/
  • 在一个单独的环境中,能使用pip就尽量使用pip,实在有问题的情况,例如没有合适的编译好的系统版本的安装包,再使用conda进行安装,不要来回混淆

2. CUDA是否要安装

  • 如果只需要训练、简单推理,则无需单独安装CUDA,直接安装pytorch;如果有部署需求,例如导出TensorRT模型,则需要进行CUDA安装

3. Pytorch安装注意事项

  • 16XX的显卡,安装cu102的版本,否则可能训练出现问题
  • 30XX、40XX显卡,要安装cu111以上的版本,否则无法运行

查询最大支持的扩大版本CUDA:11.7 image.png

image.png

三、ultralytics(YOLOv8)安装

1、直接使用源码(不是很推荐,无法使用命令行工具)

image.png

2、pip直接安装(官方推荐,个人不是很推荐)

pip install ultralytics

3、pip源码安装(个人推荐)

https://github.com/ultralytics/ultralytics && unzip ultralytics-main.zip && cd ultralytics-main
pip install -e .  (-e参数必须要有,否则后续修改代码无效)

第三种方式:后续代码修改可以检测到 image.png

四、正确使用windows中的终端

1、Windows中的终端

  • cmd (使用这个)
  • anaconda prompt (使用这个)
  • powershell (不要用!无法激活环境)

2、cmd:

conda init cmd.exe

3、在其他软件,如Pycharm、vscode中也要注意!

image.png