1、安装python  3.6  以上版本

2、安装Pycharm

3、yolov5源码 下载：GitHub下载

4、计算平台

1）CUDA（Compute Unified Device Architecture），是显卡厂商NVIDIA推出的运算平台

2）Anaconda指的是一个开源的Python发行版本，其包含了conda、Python等180多个科学包及其依赖项。 [1] 因为包含了大量的科学包

3）nvidia-smi

+-----------------------------------------------------------------------------+
| NVIDIA-SMI 442.50       Driver Version: 442.50       CUDA Version: 10.2     |
|-------------------------------+----------------------+----------------------+
| GPU  Name            TCC/WDDM | Bus-Id        Disp.A | Volatile Uncorr. ECC |
| Fan  Temp  Perf  Pwr:Usage/Cap|         Memory-Usage | GPU-Util  Compute M. |
|===============================+======================+======================|
|   0  GeForce GTX 166... WDDM  | 00000000:01:00.0 Off |                  N/A |
| N/A   40C    P8     9W /  N/A |    153MiB /  6144MiB |      0%      Default |
+-------------------------------+----------------------+----------------------+

+-----------------------------------------------------------------------------+
| Processes:                                                       GPU Memory |
|  GPU       PID   Type   Process name                             Usage      |
|=============================================================================|
|    0     10424    C+G   ...4.79.0_x64__8wekyb3d8bbwe\YourPhone.exe N/A      |
+-----------------------------------------------------------------------------+

5、安装pytorch

Start Locally | PyTorch