華文網

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

夏乙 編譯整理

量子位 出品 | 公眾號 QbitAI

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

也需要不少時間。

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

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

同時載入要訓練的模型和訓練集。

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

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

要使用Parris,你需要先準備好要訓練的模型和資料集,

安裝方式如下:

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

— 完 —

誠摯招聘