您的位置:首頁>正文

這個開源工具要自動配置機器學習環境,讓你一鍵開始訓練模型

夏乙 編譯整理

量子位 出品 | 公眾號 QbitAI

在搭建、訓練機器學習模型的過程中, 你的時間可能大部分都花在了那些與演算法無關的事情上:收集資料、清洗資料、標注, 甚至基礎環境的配置, 也需要不少時間。

有一個名叫Parris的開源工具, 就要讓配置基礎環境這個環節自動化。 今天, 它剛被發佈到Reddit論壇上就收穫了不少贊。

據作者介紹, 這個工具針對雲端的機器學習訓練, 一鍵完成配置伺服器、安裝作業系統、安裝GPU驅動、為專案安裝各種依賴包等等環節, 同時載入要訓練的模型和訓練集。

另外, 用戶還可以為伺服器設置一個自動關機時間。

Parris目前只支持亞馬遜AWS, 依靠CloudFormation範本來實現一鍵啟動訓練, 需要使用者有AWS帳戶和證書。 不過, 據作者說, 以後的版本打算支持Google雲、微軟Azure等等。

要使用Parris, 你需要先準備好要訓練的模型和資料集, 安裝方式如下:

GitHub地址:https://github.com/jgreenemi/Parris

— 完 —

誠摯招聘

同類文章
Next Article
喜欢就按个赞吧!!!
点击关闭提示