詳細Anaconda+Pycharm配置YoloV5環境筆記


環境:

  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)效果圖

 

 

七,補充:pip配置環境方法直接見官方源代碼說明文件。


免責聲明!

本站轉載的文章為個人學習借鑒使用,本站對版權不負任何法律責任。如果侵犯了您的隱私權益,請聯系本站郵箱yoyou2525@163.com刪除。



 
粵ICP備18138465號   © 2018-2025 CODEPRJ.COM