Ubuntu 20.04 安装部署 Yolo5 开发环境

目录

1.  安装显卡驱动

2.  安装 Miniconda3

3.  安装 Pytorch (GPU 版)

4.  配置 Yolo5 环境


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

1. 安装显卡驱动

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


2. 安装 Miniconda3

Ubuntu 20.04 安装部署 Yolo5 开发环境_第1张图片

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

3.  安装 Pytorch (GPU 版)

   进入 Yolo5 虚拟环境,在此虚拟环境中 安装 Pytorch GPU 版,如下所示:
Ubuntu 20.04 安装部署 Yolo5 开发环境_第2张图片
Ubuntu 20.04 安装部署 Yolo5 开发环境_第3张图片

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

4. 配置 Yolo5 环境

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

 Ubuntu 20.04 安装部署 Yolo5 开发环境_第5张图片
 

你可能感兴趣的:(Cv2,YOLO,pytorch,tensorflow2,linux)