当前位置：首页 > news >正文

X-AnyLabeling 开启 ultralytics GPU训练模式

news 2025/11/1 19:06:45

只能用源码的形式，才能开启GPU训练模式

显卡硬件（入门GTX1060 6G起步）+ 驱动 + CUDA kit + cudnn 配置CUDA环境

CUDA门道很多，

硬件算算力，决定了驱动版本，驱动版本决定了CUDA版本，CUDA版本决定了

cudnn的版本，

建议：CUDA12.4 ,即显卡驱动版本要求大于55x.xx.以及以上，对应cudnn的版本就是9.x以上

检验是否配置好

打开window power，输入命令: nvcc --version ，有如下内容表示成功！PS C:\Users\ZrSpring> nvcc --version
nvcc: NVIDIA (R) Cuda compiler driver
Copyright (c) 2005-2023 NVIDIA Corporation
Built on Wed_Nov_22_10:30:42_Pacific_Standard_Time_2024
Cuda compilation tools, release 12.x, V12.x.xxx
Build cuda_12.x.xx.x/compiler.33567101_0

配置好Python, 建议Python3.10.x -3.12.x .

打开Pycharm，点击get from VCS, 将 X-AnyLabeling克隆到本地

https://github.com/CVHub520/X-AnyLabeling.git

Pycharm项目构建完毕后，执行如下依赖

1 安装依赖pip3 install requirements-gpu-dev2 安装 pytorch （我系统使用的CUDA12.4）pip install torch==2.5.0 torchvision==0.20.0 torchaudio==2.5.0 --index-url https://download.pytorch.org/whl/cu124注意：其他匹配CUDA的版本：
在这里查阅：https://pytorch.org/get-started/previous-versions/

然后，启动X-AnyLabeling，运行文件 : X-AnyLabeling/anylabeling/app.py

训练里面就会有GPU了