Ubuntu 20.04 安装部署 Yolo5 开发环境

目录

1.  安装显卡驱动

2.  安装 Miniconda3

3.  安装 Pytorch (GPU 版)

4.  配置 Yolo5 环境

---

基于 Ubuntu20.04 安装 Yolo5 的开发环境。
 

1. 安装显卡驱动

   先安装显卡驱动。
   从 NVDIA官网上下载最新的驱动，安装后进行下一步。
   驱动下载地址：NVIDIA GeForce 驱动程序 - N 卡驱动 | NVIDIA

2. 安装 Miniconda3

       2.1  下载后进行安装。
       2.2   安装之后，使用 conda -V和 python -V查看版本信息。
       2.3   接下来，创建虚拟环境：conda create -n yolo5 python=3.8

3.  安装 Pytorch (GPU 版)

   进入 Yolo5 虚拟环境，在此虚拟环境中 安装 Pytorch GPU 版，如下所示：

   等待安装完成。 执行脚本测试 PYTORCH 和 CUDA 的功能是否正常。

4. 配置 Yolo5 环境

 去 Github 下载 Yolo5 (官网地址：https://github.com/ultralytics/yolov5/releases)，选择最新版。切换 yolo5 到虚拟环境下，cd 到 yolo5 工程目录，执行：pip install -r requirment.txt  安装依赖包，由于前面安装了 Pytorch 环境，这里需要把 requirment.txt 中的 pytorch 相关的包注释掉。