環境:
win10 20H2
需要的安裝包:
Anaconda3-2021.05-Windows-x86_64.exe
pycharm-professional-2021.2.1.exe
YoloV5源碼
CUDA10.2與CUDNN V8.2.4
pytorch 1.9
選擇性安裝OpenCV庫
一,安裝Anaconda3
下載地址:https://www.anaconda.com/products/individual-d

1.選擇CMD.exe Prompt,在命令行中輸入:
conda install git
conda install pip
2.創建Yolov5環境,選擇Python3.8,安裝完后可以關閉ANACONDA。

二,安裝pycharm
下載地址:https://www.jetbrains.com/pycharm/download/#section=windows;
路徑不能有中文空格和符號;
1.破解,我用的ide-eval-resetter-2.1.6.jar(無線30天),而jetbrains-agent.jar(現在不穩定,經常打不開pycharm):

選擇30天使用
破解教程:http://www.downxia.com/downinfo/323769.html
2.漢化:
這里就不講了,file-settings-plugins,搜索chinese,安裝中文包就可以。下載超時可以使用下載地址:https://plugins.jetbrains.com/plugin/13710-chinese-simplified-language-pack----/versions,jar包放到lib中拖到軟件里在插件里選擇從硬盤安裝就可以了。

3.Pycharm創建工程,目錄選擇YoloV5源碼文件夾(目錄無中文符號)
4.配置解釋器

結果:

三,安裝Cude10.2與CUDNN
CUDA:是顯卡廠商NVIDIA推出的運算平台。 CUDA是一種由NVIDIA推出的通用並行計算架構,該架構使GPU能夠解決復雜的計算問題。
CUDNN:NVIDIA cuDNN是用於深度神經網絡的GPU加速庫。它強調性能、易用性和低內存開銷。
詳細見:深度學習框架與運行平台
1.安裝Cude10.2
首先打開NVIDIA顯卡控制面板,查看顯卡支持的最高Cude版本,我的最高支持CUDE11.4,如下圖:

目前(2021.09.04日)pytorch支持的最新的CUDA為11.1,可是不支持Conda,如下圖,所以我們選擇CUDA10.2。

(1)下載:
下載鏈接:https://developer.nvidia.com/cuda-10.2-download-archive?target_os=Windows&target_arch=x86_64&target_version=10&target_type=exenetwork,下載主程序和相應補丁。

(2)安裝:
自定義安裝,不勾選visual studio integration

(3)安裝所有補丁:

(4)安裝驗證:
cmd中輸入nvcc -V,正確結果如下:

2.安裝CUDNN V8.2.4 For Cude10.2
(1)下載:
下載鏈接:https://developer.nvidia.com/cudnn,這里需要登錄並填寫問卷才能下載(登錄后的下載鏈接:https://developer.nvidia.com/rdp/cudnn-download)

(2)安裝:
解壓下載的cudnn-10.2-windows10-x64-v8.2.4.15.zip包,將其中的文件復制到CUDE安裝目錄中(默認為C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.2)

然后,在系統環境變量的Path中新增如下兩項(對應自己的安裝目錄)。
C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.2 C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.2\lib\x64
如下圖:

檢驗:
切換到C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.2\extras\demo_suite目錄下,shift+鼠標右鍵打開命令行 執行bandwidthTest.exe或者.\bandwidthTest.exe,查看結果是否如下圖,顯示Pass則安裝成功。

注:有什么疑問見補充:win10下CUDA和CUDNN的安裝(超詳細)!親測有效!
四,安裝Pytorch
1.安裝
打開官網https://pytorch.org/get-started/locally/,選擇使用Conda安裝,如下圖,安裝命令為:
conda install pytorch torchvision torchaudio cudatoolkit=10.2 -c pytorch

打開Anaconda Navigator — yolov5test — CMD.exe Prompt,粘貼命令,回車即可。安裝完成如下圖:

五,選擇性安裝OpenCV
打開【anaconda navigator】-yolov5test,點選下圖所示【Enviroments】,將下拉單選成【ALL】,然后搜索【opencv】,選中點擊apply完成了安裝。

六,【anaconda navigator】-yolov5test中安裝yolov5一些必須庫
torch
pyyaml
cudnn等等
詳細見:源代碼目錄下的:requirements.txt文件
+cmd中安裝pip install opencv-python和pip install thop等



六,運行項目:
1.運行Train.py
(1)結果目錄runs\train\exp

(2)提示未找到庫對應版本的不重要提示(我們安裝的版本高於它引發的提示):

(3)正確運行完成,返回代碼為0並打印輸出目錄:

(4)目錄里查看

2.運行Detect.py

(1)目錄里查看:

(2)效果圖


