公司动态

yolo26环境搭建

📅 2026/8/7 9:46:23
yolo26环境搭建
pytorch安装创建虚拟环境主要用于模型训练和模型转换因为我们要做边缘识别目标是ubuntu22404系统所以选择python10完美适配conda create-nyolo26-trainpython3.10-y查看显卡配置(export-yolo)aipaic:~/project/yolo26-train$ nvidia-smi Tue Aug414:36:032026-----------------------------------------------------------------------------------------|NVIDIA-SMI550.163.01 Driver Version:550.163.01 CUDA Version:12.4||---------------------------------------------------------------------------------------|GPU Name Persistence-M|Bus-Id Disp.A|Volatile Uncorr. ECC||Fan Temp Perf Pwr:Usage/Cap|Memory-Usage|GPU-Util Compute M.||||MIG M.||||0NVIDIA GeForce RTX4090D Off|00000000:01:00.0 Off|Off||30% 35C P8 14W / 425W|1623MiB / 24564MiB|0% Default||||N/A|--------------------------------------------------------------------------------------- -----------------------------------------------------------------------------------------|Processes:||GPU GI CI PID Type Process name GPU Memory||ID ID Usage||||0N/A N/A3407C...ip/anaconda3/envs/sw-dms/bin/python 804MiB||0N/A N/A3408C...ip/anaconda3/envs/sw-dms/bin/python 804MiB|-----------------------------------------------------------------------------------------(export-yolo)aipaic:~/project/yolo26-train$pytorch官网选定版本linux推荐pip安装https://pytorch.org/python-mpipinstalltorch2.6.0torchvision0.21.0torchaudio2.6.0 --index-url https://download.pytorch.org/whl/cu124⚠️ 重要顺序一定先固定 torch再装 ultralytics安装 ultralyticsyolo26直接锁定版本python-mpipinstallultralytics8.4.115安装其他依赖python-mpipinstallsupervision0.26.1python-mpipinstallonnx1.16.1,1.19.0python-mpipinstallonnxslim0.1.82 onnxruntimepython-mpipinstallonnxruntime-gpu