yolov4 於macOS訓練的問題

yolov4 macos pytorch

JWu 2023-06-15 10:44:29 ‧ 760 瀏覽

本身使用MacbookPro在yolov4訓練時時常卡在無法使用GPU導致訓練時間非~~常的長
經過搜尋相關內容的知MacOS不支援GPU運算

也試過Pytorch去建置訓練環境，不過還是卡在mish-cuda的調用

想請問是否有其他的方式可以調用Macbook本身的繪圖卡運算?

當然在google colab上訓練可以解決此問題
不過還是想知道有沒有macOS在local端訓練的方法?

froce iT邦大師 1 級 ‧ 2023-06-15 10:48:57 檢舉

https://www.appcoda.com.tw/coreml-introduction/
https://github.com/PROGrand/yolo-coreml

常見的cuda只有nvdia的顯卡才有。

JWu iT邦新手 5 級 ‧ 2023-06-15 11:01:30 檢舉

感謝提供網址支援!
我可以繼續研究下去了!

原本沒搞清楚用的東西是什麼

直播研討會

{{ item.channelVendor }} {{ item.webinarstarted }} |

直播中

增廣建文

iT邦研究生 5 級 ‧ 2023-06-16 09:05:19

mac想用GPU的話建議改用pytorch版，不論是M2或是AMD GPU都可以用來加速

https://github.com/Tianxiaomo/pytorch-YOLOv4/

JWu iT邦新手 5 級 ‧ 2023-06-19 17:31:15 檢舉

感謝回覆!
我有收到Pytorch版的消息，但網路上教學文章有點少

增廣建文 iT邦研究生 5 級 ‧ 2023-06-20 08:29:29 檢舉

他document應該算蠻清楚的
data格式跟原本的yolo一致

立即登入回答

參賽組數

1064 組

團體組數

40 組

累計文章數

22195 篇

完賽人數

600 人

IT邦幫忙