PyTorch+YOLOv5環境搭建（未完待續）

PyTorch+YOLOv5環境搭建

1. PyTorch requirements

   NVIDIA CUDA 9.2 or above

   NVIDIA cuDNN v7 or above

   https://github.com/pytorch/pytorch#installation

   對應的最低GPU運算能力和驅動版本爲：

   GPU Compute Capability >= 3.0

   Compatible Driver Version >= 396.26

   PyTorch版本：官方推薦版本爲Commands for Versions >= 1.0.0

   https://pytorch.org/get-started/previous-versions/

2. YOLOv5 requirements

   Python 3.7 or later

   torch >= 1.5

   https://github.com/ultralytics/yolov5

   因此

   GPU Compute Capability >= 3.0

   Compatible Driver Version >= 418.39

   NVIDIA CUDA 10.1 or above

   NVIDIA cuDNN v7.5 or above

   查詢自己GPU的Compute Capability以及支持的CUDA，cuDNN版本

   https://developer.nvidia.com/cuda-gpus

   https://developer.nvidia.com/cuda-toolkit-archive

   https://developer.nvidia.com/rdp/cudnn-archive

   https://blog.csdn.net/yumin1058882119/article/details/106900592