yolo_learn/docs/YOLO环境安装.md

参考环境

硬件：CPU（无显卡） 系统：Debian 12.6 Python: 3.8.19

INFO：Ultralytics YOLOv8.1.42 🚀 Python-3.8.19 torch-1.13.1+cu117 CPU (Intel Xeon E5-2670 v3 2.30GHz)

安装库

采用 pip 安装第三方库

pip install matplotlib==3.6.2 matplotlib-inline==0.1.6
pip install onnx==1.16.1 onnxruntime==1.16.3 netron==7.6.9 opencv-python==4.9.0.80
pip install pandas==1.5.3
pip install torch==1.13.1 torchvision==0.14.1 ultralytics==8.1.42 torchaudio==0.13.1 
# pytorch-cuda=11.7 
pip install ultralytics

因为没有显卡，所以无需安装 onnx-gpu 版本。

采用 CUDA 安装：

pip install onnx==1.16.1 onnxruntime-gpu==1.16.3
pip install tensorboard==2.8.0 tensorflow-gpu==2.8.4
pip install nvidia-cuda-nvrtc-cu11==11.7.99
pip install nvidia-cuda-runtime-cu11==11.7.99

环境验证

python -m torch.utils.collect_env

输出环境信息：

Collecting environment information...
PyTorch version: 1.13.1+cu117
Is debug build: False
CUDA used to build PyTorch: 11.7
ROCM used to build PyTorch: N/A

OS: Ubuntu 20.04.6 LTS (x86_64)
GCC version: (Ubuntu 9.4.0-1ubuntu1~20.04.2) 9.4.0
Clang version: Could not collect
CMake version: Could not collect
Libc version: glibc-2.31

Python version: 3.8.19 (default, Aug  7 2024, 15:36:56)  [GCC 9.4.0] (64-bit runtime)
Python platform: Linux-5.15.0-107-generic-x86_64-with-glibc2.29
Is CUDA available: True
CUDA runtime version: 11.4.152
CUDA_MODULE_LOADING set to: LAZY
GPU models and configuration: GPU 0: NVIDIA GeForce GTX 1060 3GB
Nvidia driver version: 550.100
cuDNN version: Probably one of the following:
/usr/local/cuda-11.4/targets/x86_64-linux/lib/libcudnn.so.8
/usr/local/cuda-11.4/targets/x86_64-linux/lib/libcudnn_adv_infer.so.8
/usr/local/cuda-11.4/targets/x86_64-linux/lib/libcudnn_adv_train.so.8
/usr/local/cuda-11.4/targets/x86_64-linux/lib/libcudnn_cnn_infer.so.8
/usr/local/cuda-11.4/targets/x86_64-linux/lib/libcudnn_cnn_train.so.8
/usr/local/cuda-11.4/targets/x86_64-linux/lib/libcudnn_ops_infer.so.8
/usr/local/cuda-11.4/targets/x86_64-linux/lib/libcudnn_ops_train.so.8
HIP runtime version: N/A
MIOpen runtime version: N/A
Is XNNPACK available: True

Versions of relevant libraries:
[pip3] numpy==1.24.4
[pip3] torch==1.13.1
[pip3] torchvision==0.14.1
[conda] Could not collect