YOLO环境配置
首页文章工具搜索

YOLO环境配置

2026.08.27 [Windows] [Python] [YOLO] [机器视觉] 190 浏览


前言

本人是在Windows下配置的YOLO环境,使用的是Ultralytics官方库。下面会详细列出配置步骤,所需要下载的内容我会在文章末尾统一列出下载链接。

确认硬件

按下Win+R,键入"cmd"打开终端,输入"nvidia-smi"。如果能看到显卡信息,请查看右上角的 CUDA Version(如下文所示),这是显卡所支持最高的CUDA版本,后续安装PyTorch时不得高于此版本!

C:\Users\heveral>nvidia-smi Mon Aug 24 22:00:00 2026 +-----------------------------------------------------------------------------------------+ | NVIDIA-SMI 610.88 KMD Version: 610.88 CUDA UMD Version: 13.3 | +-----------------------------------------+------------------------+----------------------+ | GPU Name Driver-Model | Bus-Id Disp.A | Volatile Uncorr. ECC | | Fan Temp Perf Pwr:Usage/Cap | Memory-Usage | GPU-Util Compute M. | | | | MIG M. | |=========================================+========================+======================| | 0 NVIDIA GeForce RTX 5070 ... WDDM | 00000000:01:00.0 On | N/A | | N/A 60C P4 20W / 140W | 1169MiB / 12227MiB | 6% Default | | | | N/A | +-----------------------------------------+------------------------+----------------------+

基础Python环境构建

推荐安装Python 3.8~3.11版本,安装时务必勾选"Add Python to PATH"!!!本人用的是Python 3.10(下载时请根据自己的CPU架构选择对应版本)

创建虚拟环境

这个操作可以防止YOLO的依赖包污染其他的Python项目!

# 创建名为 yolo_env 的环境(名字随意) python -m venv yolo_env # 激活环境(Windows CMD下运行) yolo_env\Scripts\activate
激活成功后,命令行前面会出现 (yolo_env) 的标志。

安装PyTorch

YOLO依赖于PyTorch,这一步是重中之重,需要根据实际需要配置设备的显卡情况选择:

情况一:有NVIDIA显卡 打开PyTorch官网,选择操作系统和CUDA版本,复制生成的命令。例如当前设备显卡驱动支持CUDA 13.2那么运行

pip3 install torch torchvision --index-url https://download.pytorch.org/whl/cu132

情况二:只有CPU无独立显卡(CPU模式运行效率奇慢无比,不推荐)请运行

pip3 install torch torchvision

验证:在终端输入"python",然后输入"import torch; print(torch.cuda.is_available())"。如果输出"True"表明GPU配置成功,反之表示在使用CPU模式。

安装Ultralytics库(YOLO)

直接使用pip安装官方整合包,他会自动下载OpenCV等依赖项。

# 使用官方渠道下载 pip install ultralytics # 如果官方渠道下载速度很感人可以 Ctrl+C 终止下载后使用这条进行下载(使用了清华大学镜像站进行加速) pip install ultralytics -i https://pypi.tuna.tsinghua.edu.cn/simple

验证环境

到这一步,环境已经配出来了,但是为了稳妥起见直接下载官方预训练模型,对一张网络图片进行检测,以此验证环境是否完全正常。

在终端(需要在虚拟环境下)执行:

yolo predict model=yolov8n.pt source='https://ultralytics.com/images/bus.jpg'

预期结果:程序会自动下载 yolov8n.pt 模型文件,然后输出检测结果,并在终端显示结果保存路径(通常在 runs/detect/predict 文件夹下)。打开该文件夹,看到标出检测框的 bus.jpg 即代表配置成功!

Python下载:链接
PyTorch下载:链接