從零開始部署pytorch+yovov4模型


之前部署了直接在win中運行的模型。考慮到很多模型都是在pytorch 中跑的,練習一下pytorch的配置過程。

1.首先有一個pytorch環境。

環境內容

torch:1.2.0
torchvision:0.4.0

Anaconda安裝

最新版本的Anaconda沒有VSCODE,如果大家為了安裝VSCODE方便可以直接安裝舊版的Anaconda,百度網盤連接如下。也可以裝新版然后分開裝VSCODE。

鏈接:https://pan.baidu.com/s/1MEdjh976CYotuQgiRWkdPA
提取碼:1234

下載好了之后,選擇合適路徑,直接安裝就好。

我選擇了Add Anaconda to my PATH environment variable,我用起來覺得更好用。

 

等待安裝完之后,就安裝了Anaconda了。

下載Cudnn和CUDA

我這里使用的是torch=1.2.0,官方推薦的Cuda版本是10.0,因此會用到cuda10.0,與cuda10.0對應的cudnn是7.4.1.5,這個組合我實驗過了,絕對是可以用的。

cude10和cudnn可以在官網下載,懶得找的我貼個百度雲連接:

鏈接:https://pan.baidu.com/s/1UCOrvg8pU4kZ2ro6JQhxQQ
提取碼:1234

下載好之后可以打開cuda_10的exe文件進行安裝。

 

這里選擇自定義。

 

 然后直接點下一步就行了。

 

 

 安裝完后在C盤這個位置可以找到根目錄。
C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.0
然后大家把Cudnn的內容進行解壓。

 

 把這里面的內容直接復制到C盤的根目錄下就可以了。

 

 

配置torch環境

Win+R啟動cmd,我沒有啟動cmd,而是啟動的anacade 的

 

 

在命令提示符內輸入以下命令:

創建環境:

conda create –n pytorch python=3.6

激活環境:

activate pytorch 

打開pytorch的官方安裝方法:
https://pytorch.org/get-started/previous-versions/
官網推薦的安裝代碼如下,我使用的是Cuda10的版本:

# CUDA 10.0
pip install torch===1.2.0 torchvision===0.4.0 -f https://download.pytorch.org/whl/torch_stable.html

# CUDA 9.2
pip install torch==1.2.0+cu92 torchvision==0.4.0+cu92 -f https://download.pytorch.org/whl/torch_stable.html

# CPU only
pip install torch==1.2.0+cpu torchvision==0.4.0+cpu -f https://download.pytorch.org/whl/torch_stable.html

需要注意的是,直接這樣安裝似乎特別慢。雖然慢,但我還是用這種慢的方式裝的,當時速度還可以。好的,繼續參考大佬博客:因此我們可以進入如下網址:
https://download.pytorch.org/whl/torch_stable.html
找到自己需要的輪子下載。

 

 

下載的時候使用迅雷下載就行了,速度還是比較快的!

下載完成后找到安裝路徑:

 

 在cmd定位過來后利用文件全名進行安裝就行了!

 

 這里我也傳一個百度網盤的版本。
鏈接: https://pan.baidu.com/s/14-QVk7Kb_CVwaVZxVPIgtw
提取碼: rg2e
全部安裝完成之后重啟電腦。

以上就是安裝pytorch的全部過程了。如何檢測自己安裝的環境是成功的呢?

打開cmd或者anacodnda prompt輸入

import torch
a = torch.cuda.is_available()
print(a)
ngpu= 1
device = torch.device("cuda:0" if (torch.cuda.is_available() and ngpu > 0) else "cpu")
print(device)
print(torch.cuda.get_device_name(0))
print(torch.rand(3,3).cuda())

輸出樣例:

True
cuda:0
GeForce GTX 1650
tensor([[0.0387, 0.8379, 0.6165],
[0.4231, 0.0060, 0.9049],
[0.7293, 0.1726, 0.9432]], device='cuda:0')

 

 

 就表示成功配置了pytorch環境。

 

安裝VSCODE

為了方便編輯代碼,需要安裝編輯器。
打開anaconda,切換環境。

 

 

 

安裝VSCODE,安裝完就可以launch一下了,之后就可以把VScode固定到任務欄上,方便打開。

當然,安裝pycharm也很好用。貼個我的安裝包:

鏈接:https://pan.baidu.com/s/1LUfAtHrYbIfhpD4dd_JsAA
提取碼:1234

 

 

 

2. 運行代碼

下載代碼

安裝好環境之后,就是運行代碼了。有GPU加持的筆記本可以進行簡單的視頻識別。我使用的代碼是github這位大佬的,大家可以來參考:

https://github.com/bubbliiiing/yolov4-pytorch#Reference

直接下載之后解壓即可。

下載完庫后解壓,在百度網盤下載yolo4_weights.pth或者yolo4_voc_weights.pth,放入model_data,

網上有很多,我發個我在用的吧:

鏈接:https://pan.baidu.com/s/1PjIfRjI8nLFL_QFgrR3QEw
提取碼:1234

 

 

打開anaconda的命令行:

#依次輸入
activate pytorch#進入pytorch環境
#進入解壓代碼的文件夾:例如:cd E:\github\yolov4-bubbliiiing\yolov4-pytorch-master
#然后運行里面的predict.py文件
python predict.py
#然后輸入需要識別的圖片路徑
img/street.jpg
#這是圖片識別;對於視頻識別和攝像頭識別,只需要打開predict.py文件進行修改即可

 


 

修改對應的命令行,即可調用視像頭或者是視頻識別。

以上就是我全部的配置過程,后續會進行訓練自己的數據,歡迎大家在留言區探討!

 


免責聲明!

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



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