yolov5 win10 CPU與GPU環(huán)境搭建過程
前言
最近實習任務為黑煙檢測，想起了可以嘗試用yolov5來跑下，之前一直都是用的RCNN系列，這次就試試yolo系列。
一、安裝pytorch
1.創(chuàng)建新的環(huán)境
打開Anaconda Prompt命令行輸入
創(chuàng)建一個新環(huán)境，并激活進入環(huán)境。
# 創(chuàng)建了名叫yolov5的，python版本為3.8的新環(huán)境 conda create -n yolov5 python=3.8 # 激活名叫yolov5的環(huán)境 conda activate yolov5
2.下載YOLOv5 github項目
下載地址為：
https://github.com/ultralytics/yolov5
如果安裝了git可以使用git clone https://github.com/ultralytics/yolov5，沒有的話直接下載zip壓縮包也行，把壓縮包解壓到指定目錄就行。
3.安裝相關依賴庫和包
查看一下requirements.txt 里面的內容并下載所有的依賴包
依次安裝，建議可以用清華源進行安裝：
先把清華源設置成默認：
pip install pip -U pip config set global.index-url https://pypi.tuna.tsinghua.edu.cn/simple
再依次安裝：
pip install tqdm pip install scipy pip install pyyaml pip install matplotlib pip install opencv-python==4.1.2.30 pip install requests pip install seaborn pip install pandas
安裝pytorch需要注意一下：
還是要先換源：
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/ conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main/ conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/conda-forge/ conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/msys2/ conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/bioconda/ conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/menpo/ conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/
然后再用下面的命令代碼確認
conda config --set show_channel_urls yes
之后進入官網https://pytorch.org/找到合適的版本，如果你是安裝cpu版
運行conda install pytorch torchvision torchaudio cpuonly
如果你是安裝GPU版
運行conda install pytorch torchvision torchaudio cudatoolkit=10.1
這里個人的計算機配置不一樣，就拿我的來說cuda10.1可以運行，cuda11無法運行GPU版，這里可以多試幾次找到適合自己的版本。
4.驗證
在剛剛建好的yolov5環(huán)境下啟動python
python #輸入庫 import torch #查看版本 print(torch.__version__)
CPU版如果到這步不報錯，就說明安裝成功了。
GPU版需要再運行一些代碼查看
#查看gpu是否可用 torch.cuda.is_available() #返回設備gpu個數(shù) torch.cuda.device_count()
一切正常的話，GPU版的pytorch就安裝成功了
二、運行detect.py文件
在建好的yolov5環(huán)境下輸入：
python detect.py
會默認下載最小的yolov5s.pt文件
檢測結果如下：
總結
總的來說還是比較簡單的，除了yolov5s.pt模型，yolov5還有
V3.1權重文件下載不下來的，可以通過下面的鏈接下載
到此這篇關于yolov5 win10 CPU與GPU環(huán)境搭建過程的文章就介紹到這了,更多相關yolov5 win10 CPU與GPU環(huán)境搭建內容請搜索腳本之家以前的文章或繼續(xù)瀏覽下面的相關文章希望大家以后多多支持腳本之家！
相關文章
這篇文章主要介紹了python中的opencv?圖像分割與提取，圖像中將前景對象作為目標圖像分割或者提取出來。對背景本身并無興趣分水嶺算法及GrabCut算法對圖像進行分割及提取。具體實現(xiàn)過程需要的朋友可以參考下面文章詳細介紹2022-06-06
pyqt5 使用setStyleSheet設置單元格的邊框樣式操作
這篇文章主要介紹了pyqt5 使用setStyleSheet設置單元格的邊框樣式操作，具有很好的參考價值，希望對大家有所幫助。一起跟隨小編過來看看吧2021-03-03
在本篇內容里小編給各位分享的是一篇關于python實現(xiàn)朋友圈中的九宮格圖片的實例講解，有需要的朋友們可以參考下。2020-09-09
Python實現(xiàn)統(tǒng)計單詞出現(xiàn)的個數(shù)
這篇文章主要介紹了Python實現(xiàn)統(tǒng)計單詞出現(xiàn)的個數(shù),本文給出了實現(xiàn)代碼以及使用方法,需要的朋友可以參考下2015-05-05
解決pip安裝報錯“error:microsoft visual c++ 14.0&nbs
這篇文章主要介紹了解決pip安裝報錯“error:microsoft visual c++ 14.0 or greater is required”問題,具有很好的參考價值,希望對大家有所幫助,如有錯誤或未考慮完全的地方,望不吝賜教2024-05-05
Matplotlib的可以把很多張圖畫到一個顯示界面,這就設計到面板切分成一個一個子圖,下面這篇文章主要給大家介紹了關于調整Matplotlib子圖大小的相關資料,文中通過實例代碼介紹的非常詳細,需要的朋友可以參考下2022-06-06
今天小編就為大家分享一篇matplotlib subplots 設置總圖的標題方法，具有很好的參考價值，希望對大家有所幫助。一起跟隨小編過來看看吧2018-05-05
Python創(chuàng)建對稱矩陣的方法示例【基于numpy模塊】
這篇文章主要介紹了Python創(chuàng)建對稱矩陣的方法,結合實例形式分析了Python基于numpy模塊實現(xiàn)矩陣運算的相關操作技巧,需要的朋友可以參考下2017-10-10
這篇文章主要介紹了Python rstrip()方法,包括rstrip方法的語法介紹和參數(shù)類型，需要的朋友可以參考下2018-11-11
這篇文章主要介紹了Python pickle模塊實現(xiàn)對象序列化,文中通過示例代碼介紹的非常詳細，對大家的學習或者工作具有一定的參考學習價值,需要的朋友可以參考下2019-11-11