記錄一次yolo環境安裝

yolo是一個object detect model,顧名思義就是拿來檢測物件的。

使用yolo官網的安裝步驟

git clone https://github.com/ultralytics/yolov5 # clone
cd yolov5
pip install -r requirements.txt # install

在全新的環境中去執行的話,可能最終安裝完成之後會變成只能夠使用CPU來運算的方式。

我解決的方式是,先將本地安裝好的pytorch刪除,然後去pytorch官網https://pytorch.org/get-started/locally/選擇適用於自己的選項來安裝。

然後再根據yolo官網的安裝步驟安裝環境,完成後,即會使用GPU來運算。速度會比CPU快得多。

 

發佈留言

發佈留言必須填寫的電子郵件地址不會公開。 必填欄位標示為 *