Ubuntu搭建GPU版本的YOLO项目环境完整教程 – 幽络源技术导航 | 全栈开发指南

Ubuntu搭建GPU版本的YOLO项目环境完整教程 – 幽络源技术导航 | 全栈开发指南

概述

在前面的步骤中,我们准备好了Cuda环境、Python环境、并已学会Python虚拟环境的创建,本节教程幽络源基于先前的环境搭建一个基于GPU版本的Yolo项目环境。

步骤1:获取Yolo官方源码

yolo官方github中下载源码,无法下载的用户可通过幽络源分享的网盘链接进行下载->ultralytics-main.zip源码

4d5f4cde-5705-462d-9451-94513f0a8323

下载后,将其放置于一个空目录中,并解压,如图

02648928-c052-42df-b448-09432713fe31

 

步骤2:获取GPU版本的torch三件套

如图官方表示获取到源码后执行 pip install ultralytic 即可下载项目全部的库,但是默认下载的torch是cpu版本的,我们需要GPU版本的,因此需先获取到GPU版本的torch

a3766c82-9a18-49f1-9002-fbcd8a239495

首先知道自己的Python版本,我是python3.9,然后还需要知道自己的cuda版本去获取对应版本的torch,在命令面板中执行

nvidia-smi

如图可以看到我的cuda版本为12.2

bd75b7e2-8958-4c28-b120-46b0663348cb

来到torch官方,由于幽络源的cuda版本为12.2,因此查找 小于等于CUDA 12.2的torch版本都可以,如图我这里选择了

#CUDA12.1
torch==2.2.0 torchvision==0.17.0 torchaudio==2.2.0

0052ace38479a6606f7b68d7ec89853b

再来到torch官方下载页,来查找适用于linux的python3.9的torch==2.2.0 torchvision==0.17.0 torchaudio==2.2.0。【若觉得下方步骤麻烦,这里幽络源已整理好torch三件套于网盘中=>linux版本torch三件套python39用】

如图在浏览器使用ctrl+F进行搜索

对于torch,我们搜索 cu121/torch-2.2.0即可快速找到,点击下载

1

同样的,搜索 cu121/torchaudio-2.2.0,cu121/torchvision-0.17.0也下载下来

3

2

下载完后将其放置于 ultralytics目录下,如图

7c82c7c9-45d3-4d50-8416-4a7ac2ff9549

步骤3:为ultralytics项目配置并进入虚拟环境

如图在ultralytics目录下执行 virtualenv venv命令为其创建虚拟环境

3a820cdf-5027-46b7-8383-ba042bbdfd2b

然后先后通过如下命令进入虚拟环境

cd venv
cd bin
source activate

96360a8e-e0e8-4619-bbb3-0a091449e964

步骤4:安装GPU版本的torch三件套

进入到虚拟环境后首先安装GPU版的torch,而不是先使用官方推荐的pip install ultralytics命令

如图幽络源首先返回到了ultralytics目录,然后执行 pip3.9 install torch-2.2.0+cu121-cp39-cp39-linux_x86_64.whl安装该torch

image

torch安装完成后,再先后执行如下命令安装 torchvision和torchaudio

pip3.9 install torchvision-0.17.0+cu121-cp39-cp39-linux_x86_64.whl
pip3.9 install torchaudio-2.2.0+cu121-cp39-cp39-linux_x86_64.whl

如图

f825e8bd-fa67-4e35-bbb9-a0769af83270

2d17be5a-7219-4273-9d4b-ee10b3ba5e0a

步骤5:安装yolo剩下的环境

由于我安装的torch-2.2.0需要numpy版本小于2,因此首先安装指定版本的numpy==1.26.4,如图

image

再执行官方的命令 pip3.9 install ultralytics

如图

image

至此yolo的所有环境配置完成,你已经可以使用该项目进行 训练模型、使用模型做目标识别任务了

结语

以上是幽络源的在Ubuntu中搭建yolo环境的详细教程,如有不懂之处可加入我们的官方QQ群307531422学习交流询问。

THE END
喜欢就支持一下吧
分享